ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix trying to upload model file as a folder, automatically package the folder

Browse Source
This commit is contained in:
allegroai 2021-01-18 11:31:37 +02:00
parent eb5d5da59e
commit c8a690569f
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/model.py
View File

@ -1139,6 +1139,15 @@ class OutputModel(BaseModel):
raise Exception('Missing a task for this model') raise Exception('Missing a task for this model')
if weights_filename is not None: if weights_filename is not None:
# Check if weights_filename is a folder, is package upload
if Path(weights_filename).is_dir():
return self.update_weights_package(
weights_path=weights_filename,
upload_uri=upload_uri,
target_filename=target_filename or Path(weights_filename).name,
auto_delete_file=auto_delete_file,
iteration=iteration)
# make sure we delete the previous file, if it exists # make sure we delete the previous file, if it exists
if self._model_local_filename != weights_filename: if self._model_local_filename != weights_filename:
delete_previous_weights_file(self._model_local_filename) delete_previous_weights_file(self._model_local_filename)