Commit cef151d5 authored by Jiangxin Dong's avatar Jiangxin Dong
Browse files

Upload New File

parent 3add120e
import torch
import data
import model
import loss
from option import args
from trainer_vd import Trainer_VD
from logger import logger
torch.manual_seed(args.seed)
chkp = logger.Logger(args)
print("Spatially-Variant MAP Model")
loader = data.Data(args)
model = model.Model(args, chkp)
loss = loss.Loss(args, chkp) if not args.test_only else None
t = Trainer_VD(args, loader, model, loss, chkp)
while not t.terminate():
t.train()
t.test()
chkp.done()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment