Alex Stoken
4 years ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
train.py
View File
| @@ -50,7 +50,7 @@ def train(hyp): | |||
| os.makedirs(wdir, exist_ok=True) | |||
| last = wdir + 'last.pt' | |||
| best = wdir + 'best.pt' | |||
| results_file = log_dir + 'results.txt' | |||
| results_file = log_dir + os.sep + 'results.txt' | |||
| epochs = opt.epochs # 300 | |||
| batch_size = opt.batch_size # 64 | |||
| @@ -303,7 +303,7 @@ def train(hyp): | |||
| model=ema.ema, | |||
| single_cls=opt.single_cls, | |||
| dataloader=testloader, | |||
| fast=epoch < epochs / 2 | |||
| fast=epoch < epochs / 2, | |||
| save_dir=log_dir) | |||
| # Write | |||
Loading…