Hi PompousBeetle71 , Trains will log all the torch.save call, I'm assuming they do not actually use it for the rest of the files on that folder.
If you like to share a code snippet we could see if we could auto-magically log it You could use artifacts and store the entire folder. It will zip it an upload it. Then you can reuse it from other experiments. https://allegro.ai/docs/task.html?highlight=artifact#trains.task.Task.upload_artifact
Example:
task.upload_artifact('transformer', './my_trans_folder')Then from another experiment you could do:
local_copy_of_the_folder = Task.get_task(task_id='11111').artifacts['transformer'].get()