Commit
·
915b148
1
Parent(s):
0b514da
default check_git_status() to True
Browse files
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 |
|