the behavior of run.log_model()
loveunk opened this issue · comments
Kevin commented
It looks like the function log_model(model)
always tries to search the paths and build a huge zip file for uploading.
Is it possible to log model without any custom_modules? Thanks.
Michael Liu commented
Thanks for the question, @loveunk
Much of the path-searching behavior can be skipped by passing an empty list as an argument:
run.log_model(my_model, custom_modules=[])
This will greatly reduce the size of the uploaded custom_modules
.
Let me know if there's anything else I can help with, here!
Kevin commented
@convoliution Cool, thanks for your help.