Method save

models/strand_codec.py:401–412

(self, root_folder, experiment_name, hyperparams, iter_nr, info=None)

       
                                                     
    def save(self, root_folder, experiment_name, hyperparams, iter_nr, info=None):
        name=str(iter_nr)
        if info is not None:
            name+="_"+info
        models_path = os.path.join(root_folder, experiment_name, name, "models")
        if not os.path.exists(models_path):
            os.makedirs(models_path, exist_ok=True)
        torch.save(self.state_dict(), os.path.join(models_path, "strand_codec.pt"))

        hyperparams_params_path=os.path.join(models_path, "hyperparams.json")
        with open(hyperparams_params_path, 'w', encoding='utf-8') as f:
            json.dump(vars(hyperparams), f, ensure_ascii=False, indent=4)


    def forward(self, gt_dict, hyperparams, normalization_dict):