# The library
import glob
def GetLastCheckpointName(pattern: str) :
# Getting files with pattern and sorting
checkpoints = sorted(glob.glob(pattern))
# return None if there is no file
if len(checkpoints) == 0:
return None
# return last element of array
return checkpoints[len(checkpoints) - 1]
lastcp = GetLastCheckpointName("generator_checkpoint*.pth.tar")
1条答案
按热度按时间chy5wohz1#
据我所知,您尝试打开“generator_checkpoint19.pth.tar”以在代码中加载tensorflow/keras检查点,但该文件不存在。3种修复方法:
1
如果您直接输入文件名来加载检查点,请检查它。
二
如果使用循环加载(或仅检查)多个检查点,请使用
检查是否有文件。
三
如果你有多个检查点,你想得到最后一个。使用“glob”库来得到检查点。