소스 검색
return `opt` from `train.run()` (#6581)
modifyDataloader
Christoph Fürbacher
GitHub
2 년 전
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1개의 변경된 파일과
1개의 추가작업 그리고
0개의 파일을 삭제
-
train.py
|
|
@@ -634,6 +634,7 @@ def run(**kwargs): |
|
|
|
for k, v in kwargs.items(): |
|
|
|
setattr(opt, k, v) |
|
|
|
main(opt) |
|
|
|
return opt |
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |