Fix style

This commit is contained in:
Eren Gölge 2021-08-30 13:12:13 +00:00
parent 5255e089e6
commit 738eee0cf9
1 changed files with 1 additions and 1 deletions

View File

@ -328,7 +328,7 @@ class Trainer:
return obj
print(" > Restoring from %s ..." % os.path.basename(restore_path))
checkpoint = load_fsspec(restore_path, map_location='cpu')
checkpoint = load_fsspec(restore_path, map_location="cpu")
try:
print(" > Restoring Model...")
model.load_state_dict(checkpoint["model"])