Commit
·
86784cf
1
Parent(s):
0a08375
--resume bug fix #252
Browse files
train.py
CHANGED
@@ -378,7 +378,7 @@ if __name__ == '__main__':
|
|
378 |
parser.add_argument('--multi-scale', action='store_true', help='vary img-size +/- 50%')
|
379 |
parser.add_argument('--single-cls', action='store_true', help='train as single-class dataset')
|
380 |
opt = parser.parse_args()
|
381 |
-
opt.weights = last if opt.resume else opt.weights
|
382 |
opt.cfg = check_file(opt.cfg) # check file
|
383 |
opt.data = check_file(opt.data) # check file
|
384 |
print(opt)
|
|
|
378 |
parser.add_argument('--multi-scale', action='store_true', help='vary img-size +/- 50%')
|
379 |
parser.add_argument('--single-cls', action='store_true', help='train as single-class dataset')
|
380 |
opt = parser.parse_args()
|
381 |
+
opt.weights = last if opt.resume and not opt.weights else opt.weights
|
382 |
opt.cfg = check_file(opt.cfg) # check file
|
383 |
opt.data = check_file(opt.data) # check file
|
384 |
print(opt)
|