From 1f470929c037af49b3c6047e65c30ce7731417e2 Mon Sep 17 00:00:00 2001 From: krisbock Date: Thu, 16 Apr 2020 08:24:38 +1000 Subject: [PATCH] add upload_file to fix issue #56 --- examples/deep_predictive_maintenance/train/entry.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/deep_predictive_maintenance/train/entry.py b/examples/deep_predictive_maintenance/train/entry.py index 26e2662..a3c0792 100644 --- a/examples/deep_predictive_maintenance/train/entry.py +++ b/examples/deep_predictive_maintenance/train/entry.py @@ -74,4 +74,6 @@ model_path = os.path.join(output_dir, 'network.pth') torch.save(network, model_path) + # child runs have to upload model to Experiment before register will work + run.upload_file(model_path, model_path) run.register_model(model_name = 'network.pth', model_path = model_path) \ No newline at end of file