ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 15:45:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix model package was not recursive when packaging a folder

Browse Source
This commit is contained in:
allegroai 2020-05-13 20:31:31 +03:00
parent 19a24c4b20
commit 99f987aafb
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

@ -1201,7 +1201,7 @@ class OutputModel(BaseModel):
raise ValueError('Model update weights package should get either directory path to pack or a list of files') raise ValueError('Model update weights package should get either directory path to pack or a list of files')
if not weights_filenames: if not weights_filenames:
weights_filenames = list(map(six.text_type, Path(weights_path).glob('*'))) weights_filenames = list(map(six.text_type, Path(weights_path).rglob('*')))
# create packed model from all the files # create packed model from all the files
fd, zip_file = mkstemp(prefix='model_package.', suffix='.zip') fd, zip_file = mkstemp(prefix='model_package.', suffix='.zip')