updates
This commit is contained in:
parent
06e274f7e4
commit
d2ef817b1f
1
train.py
1
train.py
|
@ -369,7 +369,6 @@ if __name__ == '__main__':
|
||||||
parser.add_argument('--weights', type=str, default='', help='initial weights') # i.e. weights/darknet.53.conv.74
|
parser.add_argument('--weights', type=str, default='', help='initial weights') # i.e. weights/darknet.53.conv.74
|
||||||
opt = parser.parse_args()
|
opt = parser.parse_args()
|
||||||
opt.weights = 'weights/last.pt' if opt.resume else opt.weights
|
opt.weights = 'weights/last.pt' if opt.resume else opt.weights
|
||||||
opt.transfer = True
|
|
||||||
print(opt)
|
print(opt)
|
||||||
|
|
||||||
tb_writer = None
|
tb_writer = None
|
||||||
|
|
Loading…
Reference in New Issue