Add git info to cls, seg checkpoints (#10217)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-11-19 03:32:21 +01:00
1 parent 9286336cb4
commit 0307954e4e
3 files changed
+5 -9

No files matched your search

+1 -1
View File
@@ -376,7 +376,7 @@ def train(hyp, opt, device, callbacks): # hyp is path/to/hyp.yaml or hyp dictio
'updates': ema.updates,
'optimizer': optimizer.state_dict(),
'opt': vars(opt),
'git': GIT,
'git': GIT, # {remote, branch, commit} if a git repo
'date': datetime.now().isoformat()}
# Save last, best and delete