Loading request...
User encounters an error when trying to resume training from `checkpoint_best.pth` due to missing keys in `state_dict`. Suggests changing `load_state_dict` to `strict=False` to enable more flexible checkpoint loading.
When I try to feed in a `checkpoint_best.pth` from a previous run to resume training from checkpoint, I get an error at this line (https://github.com/salesforce/LAVIS/blob/main/lavis/runners/runner_base.py#L628) on missing keys in state_dict. I think the fix is to change the line to `self.unwrap_dist_model(self.model).load_state_dict(state_dict, strict=False)`. Thanks!