Skip to content
This repository was archived by the owner on Jun 4, 2025. It is now read-only.

Commit 6d3667a

Browse files
committed
Update: remove model versioning
1 parent 4b4b932 commit 6d3667a

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

export.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -440,14 +440,11 @@ def create_checkpoint(epoch, model, optimizer, ema, sparseml_wrapper, **kwargs):
440440
if not pickle:
441441
ckpt_model = ckpt_model.state_dict()
442442

443-
version = 6 if isinstance([module for module in model.model.modules()][1], Conv) else 5
444-
445443
return {'epoch': epoch,
446444
'model': ckpt_model,
447445
'optimizer': optimizer.state_dict(),
448446
'yaml': yaml,
449447
'hyp': model.hyp,
450-
'version': version,
451448
**ema.state_dict(pickle),
452449
**sparseml_wrapper.state_dict(),
453450
**kwargs}

0 commit comments

Comments
 (0)