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

ultralytics 8.0.176 update Dockerfile-arm64 to 22.04 (#4857)

Browse source 
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kalen Michael <kalenmike@gmail.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-09-12 17:56:37 +02:00 • committed by GitHub
parent fdf08d823e
commit e73447effb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 29 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ultralytics/hub/session.py
View file

@ -117,8 +117,7 @@ class HUBTrainingSession:
if data['status'] == 'new': # new model to start training
self.train_args = {
# TODO deprecate 'batch_size' argument in favor of 'batch'
'batch': data['batch' if 'batch' in data else 'batch_size'],
'batch': data['batch_size'], # note HUB argument is slightly different
'epochs': data['epochs'],
'imgsz': data['imgsz'],
'patience': data['patience'],
@ -159,6 +158,7 @@ class HUBTrainingSession:
data = {'epoch': epoch}
if final:
data.update({'type': 'final', 'map': map})
filesize = Path(weights).stat().st_size
smart_request('post',
url,
data=data,
@ -167,7 +167,7 @@ class HUBTrainingSession:
retry=10,
timeout=3600,
thread=False,
progress=True,
progress=filesize,
code=4)
else:
data.update({'type': 'epoch', 'isBest': bool(is_best)})