sparseml icon indicating copy to clipboard operation
sparseml copied to clipboard

Published 20 hours ago • verified publisher icon neuralmagic

don't save epoch on IC one_shot checkpoints

Open bfineran opened this issue 3 years ago • 1 comments

default saved epoch for one_shot in the IC flows is -1 due to Trainer initialization. This will cause issues on model load since the checkpoint recipe will be initialized to epoch -1, not applying any of the optimizations. This PR removes saving the epoch for one shot models so the entire one shot recipe will be applied on checkpoint load.

test_plan: @anmarques to verify

bfineran avatar Aug 12 '22 15:08 bfineran

@anmarques @rahul-tuli assigned for review

github-actions[bot] avatar Aug 12 '22 15:08 github-actions[bot]