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()