From 915b1481fc6a2454c518867ee0106a37e80e6876 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 15 Jun 2020 16:18:46 -0700 Subject: [PATCH] default check_git_status() to True --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 33fa8af..93802c5 100644 --- a/train.py +++ b/train.py @@ -363,6 +363,7 @@ def train(hyp): if __name__ == '__main__': + check_git_status() parser = argparse.ArgumentParser() parser.add_argument('--epochs', type=int, default=300) parser.add_argument('--batch-size', type=int, default=16) @@ -389,7 +390,6 @@ if __name__ == '__main__': print(opt) 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) - # check_git_status() if device.type == 'cpu': mixed_precision = False