Do you mean to simply hide or solve this warning? If it is hidden,
import warnings with warnings.catch_warnings(): warnings.simplefilter("ignore", UserWarning) model.load_state_dict(state_dict, strict=False)