本文整理汇总了Python中Engine.Engine.get_epoch_model方法的典型用法代码示例。如果您正苦于以下问题:Python Engine.get_epoch_model方法的具体用法?Python Engine.get_epoch_model怎么用?Python Engine.get_epoch_model使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类Engine.Engine
的用法示例。
在下文中一共展示了Engine.get_epoch_model方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: initBase
# 需要导入模块: from Engine import Engine [as 别名]
# 或者: from Engine.Engine import get_epoch_model [as 别名]
def initBase(configfile=None, targetMode=None, epoch=None):
"""
:type configfile: str | None
"""
global isInitialized
isInitialized = True
# Run through in any case. Maybe just to set targetMode.
global config
if not config:
if configfile is None:
configfile = DefaultSprintCrnnConfig
assert os.path.exists(configfile)
rnn.initThreadJoinHack()
rnn.initConfig(configfile, [])
config = rnn.config
rnn.initLog()
rnn.initConfigJsonNetwork()
if targetMode:
setTargetMode(targetMode)
initDataset()
if targetMode and targetMode == "forward" and epoch:
model_filename = config.value('model', '')
fns = [Engine.epoch_model_filename(model_filename, epoch, is_pretrain) for is_pretrain in [False, True]]
fns_existing = [fn for fn in fns if os.path.exists(fn)]
assert len(fns_existing) == 1, "%s not found" % fns
model_epoch_filename = fns_existing[0]
config.set('load', model_epoch_filename)
assert Engine.get_epoch_model(config)[1] == model_epoch_filename
global engine
if not engine:
devices = rnn.initDevices()
rnn.printTaskProperties(devices)
rnn.initEngine(devices)
engine = rnn.engine
assert isinstance(engine, Engine)