updates
This commit is contained in:
parent
345f4773b7
commit
caeba13b84
2
train.py
2
train.py
|
@ -95,7 +95,7 @@ def main(opt):
|
||||||
|
|
||||||
# Multi-Scale Training
|
# Multi-Scale Training
|
||||||
# img_size = random.choice(range(10, 20)) * 32
|
# img_size = random.choice(range(10, 20)) * 32
|
||||||
# dataloader = ListDataset(train_path, batch_size=opt.batch_size, img_size=img_size)
|
# dataloader = load_images_and_labels(train_path, batch_size=opt.batch_size, img_size=img_size, augment=True)
|
||||||
# print('Running this epoch with image size %g' % img_size)
|
# print('Running this epoch with image size %g' % img_size)
|
||||||
|
|
||||||
# Update scheduler
|
# Update scheduler
|
||||||
|
|
Loading…
Reference in New Issue