Browse Source

return `opt` from `train.run()` (#6581)

modifyDataloader
Christoph Fürbacher GitHub 2 years ago
parent
commit
9cf80b7f60
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      train.py

+ 1
- 0
train.py View File

@@ -634,6 +634,7 @@ def run(**kwargs):
for k, v in kwargs.items():
setattr(opt, k, v)
main(opt)
return opt


if __name__ == "__main__":

Loading…
Cancel
Save