Ejemplo n.º 1
0
    def ListSavedModels(self):
        sModelFolders = []
        if Storage.IsExistingPath(self.ExperimentModelFolder):
            if not Storage.IsFolderEmpty(self.ExperimentModelFolder):
                sModelFolders = Storage.GetDirectoriesSorted(
                    self.ExperimentModelFolder)

        oModels = []
        for sModel in sModelFolders:
            sFolder = Storage.JoinPath(self.ExperimentModelFolder, sModel)
            sModelFiles = Storage.GetFilesSorted(sFolder)
            nEpochNumber = int(sModel)
            oModels.append([nEpochNumber, sFolder, sModelFiles])

        return oModels
Ejemplo n.º 2
0
    def __init__(self, p_oParent, p_nFoldNumber, p_bIsRun=False):
        #........ |  Instance Attributes | ..............................................
        self.ParentExperiment = p_oParent
        self.FoldNumber = p_nFoldNumber
        self.IsRun = p_bIsRun
        if self.IsRun:
            self.Folder = os.path.join(self.ParentExperiment.RunBaseFolder,
                                       "fold%.2d" % self.FoldNumber)
        else:
            self.Folder = os.path.join(self.ParentExperiment.BaseFolder,
                                       "fold%.2d" % self.FoldNumber)
        Storage.EnsurePathExists(self.Folder)

        sFolders = Storage.GetDirectoriesSorted(self.Folder)
        if len(sFolders) > 0:
            self.LastUID = sFolders[-1]
        else:
            self.LastUID = ExperimentSubFolder.NO_SUBFOLDERS

        self.__pathsToEnsure = None