def load(load_dir, epoch):
with (load_dir/meta_name).open('rb') as f:
storage = Storage(*np.load(f)[0])
serializers.load_npz(
str(load_dir/model_name(epoch)),
storage.model
)
serializers.load_npz(
str(load_dir/optimizer_name(epoch)),
storage.optimizer
)
return storage
评论列表
文章目录