models/Transformer/class_wrapper.py
def save(self, ckpt_dir=None):
        """
        Saving the model to the current check point folder with name best_model_forward.pt
        :return: None
        """
        if ckpt_dir is None:
            ckpt_dir = self.ckpt_dir
        # torch.save(self.model.state_dict, os.path.join(self.ckpt_dir, 'best_model_state_dict.pt'))
        torch.save(self.model, os.path.join(ckpt_dir, 'best_model_forward.pt'))

        Saving the model to the current check point folder with name best_model_forward.pt
        :return: None
        
Saving the model to the current check point folder with name best_model_forward.pt
Saving the model to the current check point folder with name best_model_forward . pt
def save(self, ckpt_dir=None):
        if ckpt_dir is None:
            ckpt_dir = self.ckpt_dir
        torch.save(self.model, os.path.join(ckpt_dir, 'best_model_forward.pt'))
def save ( self , ckpt_dir = None ) : if ckpt_dir is None : ckpt_dir = self . ckpt_dir torch . save ( self . model , os . path . join ( ckpt_dir , 'best_model_forward.pt' ) )
Saving the model to the current check point folder with name best_model_forward.pt
Saving the model to the current check point folder with name best_model_forward . pt
"""
        Saving the model to the current check point folder with name best_model_forward.pt
        :return: None
        """ # torch.save(self.model.state_dict, os.path.join(self.ckpt_dir, 'best_model_state_dict.pt'))

None
self
ckpt_dir