how to use pytorch and caffe built by ourselves？

Question

Eric-Zhang1990 opened this issue 6 years ago · 2 comments

@animeshsingh Can you tell me how to use framework like pytorch and caffe built by ourselves? Thanks very much!

Answer 1 · 2019-01-14T19:21:13.000Z

Hi @Eric-Zhang1990. To use your own framework in FfDL, simply change the framework fields in your manifest file.

framework:
  name: custom
  version: "docker.io/namespace/imagename:version"
  command: <Your entrypoint command>

Answer 2 · 2019-01-15T01:22:40.000Z

@Tomcli Ok, thank you for your kind reply.