pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

Resuming compiled model checkpoint

Open mjamroz opened this issue 8 months ago • 2 comments

Sometimes im getting Missing key(s) in state_dict: "stem.conv1.c.weight", [...] Unexpected key(s) in state_dict: "_orig_mod.stem.conv1.c.weight", [...] while trying to resume training from checkpoint of compiled (--torchcompile=inductor) model. This PR solves the issue by stripping _orig_mod. prefix before loading weights.

mjamroz avatar Oct 23 '23 07:10 mjamroz