def setup_model(args):
if args.model_type.lower() == "lstm":
model = LSTM(args)
else:
print('set valid model type name')
exit()
optimizer = model.setup_optimizer()
if args.gpu >= 0:
cuda.get_device(args.gpu).use()
model.to_gpu()
return model, optimizer
train_model.py 文件源码
python
阅读 23
收藏 0
点赞 0
评论 0
评论列表
文章目录