Skip to content

a little wrong~ #25

@niuniuBUAA

Description

@niuniuBUAA
      torch.save(mod_chkpt, self.model_path + 'weights/best-model_epoch-{:03}_mae-{:.4f}_loss-{:.4f}.pth'.format(epoch, best_test_mae, te_avg_loss))
     while
     self.model_path = args.base_save_path + '/alph-{}_wbce_w0-{}_w1-{}'.format(str(self.alpha_sal), str(self.wbce_w0), str(self.wbce_w1))
     you are supposed to plus a '/'  in   self.model_path = args.base_save_path + '/alph-{}_wbce_w0-{}_w1-{}/'.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions