Skip to content

Commit c1878a6

Browse files
Merge pull request #792 from analysiscenter/fix/load-model
Fix load
2 parents 7703554 + 1921a8c commit c1878a6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

batchflow/models/torch/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ def callable_init(module): # example of a callable for init
405405

406406
def __init__(self, config=None):
407407
if config is not None and not isinstance(config, (dict, Config)):
408-
config = {'load/file': config}
408+
config = {'load/path': config}
409409
self.model_lock = Lock()
410410

411411
# Configs

0 commit comments

Comments
 (0)