Upload New File

main.py

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