VertaAI / modeldb

Open Source ML Model Versioning, Metadata, and Experiment Management

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

the behavior of run.log_model()

loveunk opened this issue · comments

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.

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!

commented

@convoliution Cool, thanks for your help.