I have a question on basic understanding for which i do not find answer in the doc.
If i understand well, the documentation says that we do not have to anything if we use the AWS deep learning containers.
How is the association between the hook and the network is done?
Does it use the object returned by model_fn and expect to be a mxnet module, or pytorch or tentorflow model?
Actually my model_fn() function returns an object which encapsulates my mxnet module, will it work for me, or should i create especifically the hooks , etc.. ?
Thank you
I have a question on basic understanding for which i do not find answer in the doc.
If i understand well, the documentation says that we do not have to anything if we use the AWS deep learning containers.
How is the association between the hook and the network is done?
Does it use the object returned by model_fn and expect to be a mxnet module, or pytorch or tentorflow model?
Actually my model_fn() function returns an object which encapsulates my mxnet module, will it work for me, or should i create especifically the hooks , etc.. ?
Thank you