iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update LFNA with resume

Browse Source
This commit is contained in:
D-X-Y
2021-05-17 04:33:40 +00:00
parent b11cfe263d
commit de8cf677d9
3 changed files with 62 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/xlayers/super_module.py
View File

@@ -3,6 +3,7 @@
#####################################################
import os
from pathlib import Path
import abc
import tempfile
import warnings
@@ -90,6 +91,10 @@ class SuperModule(abc.ABC, nn.Module):
total += buf.numel()
return total
def set_best_dir(self, xdir):
self._meta_info[BEST_DIR_KEY] = str(xdir)
Path(xdir).mkdir(parents=True, exist_ok=True)
def save_best(self, score):
if BEST_DIR_KEY not in self._meta_info:
tempdir = tempfile.mkdtemp("-xlayers")
@@ -97,7 +102,7 @@ class SuperModule(abc.ABC, nn.Module):
if BEST_SCORE_KEY not in self._meta_info:
self._meta_info[BEST_SCORE_KEY] = None
best_score = self._meta_info[BEST_SCORE_KEY]
if best_score is None or best_score < score:
if best_score is None or best_score <= score:
best_save_path = os.path.join(
self._meta_info[BEST_DIR_KEY],
"best-{:}.pth".format(self.__class__.__name__),