Where should the model be saved to be loaded by aicrowd_helpers.submit()

The pytorch and tensorflow sample codes save their models in different directories:

  • scratch/shared/MODEL_NAME/representation/pytorch_model.pt
  • scratch/shared/MODEL_NAME/tfhub

Therefore, I’m a bit confused how the evaluator of aicrowd decides to load the models given that there is no environment variable to define this. Will it follow the same logic as the evaluate.py file?

@amirabdi: We do provide the path to save the models here :

And in case of pytorch, they are being picked up from here :

1 Like