This commit is contained in:
Glenn Jocher 2018-09-01 19:13:12 +02:00
parent f823b3f122
commit eeb546ed6f
1 changed files with 2 additions and 2 deletions

View File

@ -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: