Add git info to cls, seg checkpoints (#10217)
This commit is contained in:
1 parent
9286336cb4
commit
0307954e4e
3 files changed
+5
-9
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user