-
Notifications
You must be signed in to change notification settings - Fork 49
Open
Description
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
Labels
No labels