glenn-jocher commited on
Commit
915b148
·
1 Parent(s): 0b514da

default check_git_status() to True

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -363,6 +363,7 @@ def train(hyp):
363
 
364
 
365
  if __name__ == '__main__':
 
366
  parser = argparse.ArgumentParser()
367
  parser.add_argument('--epochs', type=int, default=300)
368
  parser.add_argument('--batch-size', type=int, default=16)
@@ -389,7 +390,6 @@ if __name__ == '__main__':
389
  print(opt)
390
  opt.img_size.extend([opt.img_size[-1]] * (2 - len(opt.img_size))) # extend to 2 sizes (train, test)
391
  device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size)
392
- # check_git_status()
393
  if device.type == 'cpu':
394
  mixed_precision = False
395
 
 
363
 
364
 
365
  if __name__ == '__main__':
366
+ check_git_status()
367
  parser = argparse.ArgumentParser()
368
  parser.add_argument('--epochs', type=int, default=300)
369
  parser.add_argument('--batch-size', type=int, default=16)
 
390
  print(opt)
391
  opt.img_size.extend([opt.img_size[-1]] * (2 - len(opt.img_size))) # extend to 2 sizes (train, test)
392
  device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size)
 
393
  if device.type == 'cpu':
394
  mixed_precision = False
395