updates
This commit is contained in:
parent
f823b3f122
commit
eeb546ed6f
4
train.py
4
train.py
|
@ -93,10 +93,10 @@ def main(opt):
|
||||||
for epoch in range(opt.epochs):
|
for epoch in range(opt.epochs):
|
||||||
epoch += start_epoch
|
epoch += start_epoch
|
||||||
|
|
||||||
# Random input
|
# 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 = ListDataset(train_path, batch_size=opt.batch_size, img_size=img_size)
|
||||||
# print('Running image size %g' % img_size)
|
# print('Running this epoch with image size %g' % img_size)
|
||||||
|
|
||||||
# Update scheduler
|
# Update scheduler
|
||||||
# if epoch % 25 == 0:
|
# if epoch % 25 == 0:
|
||||||
|
|
Loading…
Reference in New Issue