提交 915b1481 authored 作者: Glenn Jocher's avatar Glenn Jocher

default check_git_status() to True

上级 0b514dac
...@@ -363,6 +363,7 @@ def train(hyp): ...@@ -363,6 +363,7 @@ def train(hyp):
if __name__ == '__main__': if __name__ == '__main__':
check_git_status()
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--epochs', type=int, default=300) parser.add_argument('--epochs', type=int, default=300)
parser.add_argument('--batch-size', type=int, default=16) parser.add_argument('--batch-size', type=int, default=16)
...@@ -389,7 +390,6 @@ if __name__ == '__main__': ...@@ -389,7 +390,6 @@ if __name__ == '__main__':
print(opt) print(opt)
opt.img_size.extend([opt.img_size[-1]] * (2 - len(opt.img_size))) # extend to 2 sizes (train, test) opt.img_size.extend([opt.img_size[-1]] * (2 - len(opt.img_size))) # extend to 2 sizes (train, test)
device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size) device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size)
# check_git_status()
if device.type == 'cpu': if device.type == 'cpu':
mixed_precision = False mixed_precision = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论