import utility
import data
import model
import loss
from option import args
from checkpoint import Checkpoint
from trainer import Trainer
utility.set_seed(args.seed) # 设置随机种子,方便结果复现
checkpoint = Checkpoint(args)
if checkpoint.ok:
loader = data.Data(args)
model = model.Model(args, checkpoint)
loss = loss.Loss(args, checkpoint) if not args.test_only else None
t = Trainer(args, loader, model, loss, checkpoint)
while not t.terminate():
t.train()
t.test()
checkpoint.done()
