import torch
import utility
import data
import model
import loss
from option import args
from trainer import Trainer
# To make sure that each time the-generade-random-number is fixed
torch.manual_seed(args.seed)
#print ("1-1-checkpoint")
# Open the module utility
checkpoint = utility.checkpoint(args)
#print ("1-2-checkpoint")
#
if checkpoint.ok:
print ("1-3-checkpoint")
loader = data.Data(args,None)
#print ("1-4-checkpoint")
model = model.Model(args, checkpoint)
#print ("1-5-checkpoint")
loss = loss.Loss(args, checkpoint) if not args.test_only else None
#print ("1-6-checkpoint")
t = Trainer(args, loader, model, loss, checkpoint)
#print ("1-7-checkpoint")
while not t.terminate():
t.train()
t.test()
#print ("1-8a-checkpoint")
checkpoint.done()
print ("1-8-checkpoint")main.py
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
- 文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
推荐阅读更多精彩内容
- 原文: http://zengrong.net/post/2192.htm 本站文章除注明转载外，均为本站原创或...
- 第一组：杨昊 有关表格的样式设定问题——固定头、列 前几天接到的任务中，有这样一个需求：就是表单的头部、首列需要...
