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