HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Simplify argument names (#141)

Browse source
This commit is contained in:
Glenn Jocher 2023-01-03 19:47:56 +01:00 • committed by GitHub
parent d74de2582c
commit 3c18aa5278
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 22 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/engine/trainer.py
View file

@ -102,7 +102,7 @@ class BaseTrainer:
yaml_save(self.save_dir / 'args.yaml', OmegaConf.to_container(self.args, resolve=True)) # save run args
self.last, self.best = self.wdir / 'last.pt', self.wdir / 'best.pt' # checkpoint paths
self.batch_size = self.args.batch_size
self.batch_size = self.args.batch
self.epochs = self.args.epochs
self.start_epoch = 0
if RANK == -1: