updates
This commit is contained in:
parent
65228ba8a2
commit
119d39599e
2
train.py
2
train.py
|
@ -36,7 +36,7 @@ def main(opt):
|
||||||
if platform == 'darwin': # macos
|
if platform == 'darwin': # macos
|
||||||
train_path = data_config['valid']
|
train_path = data_config['valid']
|
||||||
else: # linux (gcp cloud)
|
else: # linux (gcp cloud)
|
||||||
train_path = '../coco/trainvalno5k.txt'
|
train_path = '../coco/trainvalno5k.part'
|
||||||
|
|
||||||
# Initialize model
|
# Initialize model
|
||||||
model = Darknet(opt.cfg, opt.img_size)
|
model = Darknet(opt.cfg, opt.img_size)
|
||||||
|
|
Loading…
Reference in New Issue