I was able to work around it by overloading load_state_dict() with the following:
    def load_state_dict(self, state_dict, strict=True):
self.train_loss_data.resize_(state_dict[‘train_loss_data’].shape)
super().load_state_dict(state_dict, strict)