From 90553bac01a4fe16f6078faec94437f33a785a63 Mon Sep 17 00:00:00 2001
From: BobYeah <635596704@qq.com>
Date: Mon, 18 Jan 2021 23:11:36 +0800
Subject: [PATCH] sync
---
run_spherical_view_syn.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/run_spherical_view_syn.py b/run_spherical_view_syn.py
index 5f2e346..ea5df34 100644
--- a/run_spherical_view_syn.py
+++ b/run_spherical_view_syn.py
@@ -7,7 +7,7 @@ from tensorboardX import SummaryWriter
from torch import nn
sys.path.append(os.path.abspath(sys.path[0] + '/../'))
-__package__ = "deep_view_syn"
+__package__ = "deeplightfield"
parser = argparse.ArgumentParser()
parser.add_argument('--device', type=int, default=3,
@@ -138,7 +138,7 @@ def train_loop(data_loader, optimizer, loss, perf, writer, epoch, iters):
loss_min = 1e5
loss_max = 0
loss_avg = 0
- perf = SimplePerf(opt.simple_log)
+ perf1 = SimplePerf(opt.simple_log, True)
for _, gt, rays_o, rays_d in data_loader:
patch = (len(gt.size()) == 4)
gt = gt.to(device.GetDevice())
@@ -185,7 +185,7 @@ def train_loop(data_loader, optimizer, loss, perf, writer, epoch, iters):
iters += 1
sub_iters += 1
if opt.simple_log:
- perf.Checkpoint('Epoch %d (%.2e/%.2e/%.2e)' % (epoch, loss_min, loss_avg, loss_max), True)
+ perf1.Checkpoint('Epoch %d (%.2e/%.2e/%.2e)' % (epoch, loss_min, loss_avg, loss_max), True)
return iters
--
GitLab