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 update_weights() to use model upload target file when passed

Browse Source
This commit is contained in:
allegroai 2020-04-13 19:00:35 +03:00
parent 4b9c5c235c
commit 12659307a8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trains/model.py
View File

@ -1056,7 +1056,7 @@ class OutputModel(BaseModel):
# select the correct file extension based on the framework, or update the framework based on the file extension
framework, file_ext = Framework._get_file_ext(
framework=self._get_model_data().framework,
filename=weights_filename or register_uri
filename=target_filename or weights_filename or register_uri
)
if weights_filename: