From 3ed182299c95a1c87f9c6cb340f854315f79ee1d Mon Sep 17 00:00:00 2001
From: Zhewei Huang <598460606@163.com>
Date: Thu, 6 Jan 2022 17:54:52 +0800
Subject: [PATCH] Update train.py

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index b5c5479..93a4dc4 100644
--- a/train.py
+++ b/train.py
@@ -119,7 +119,7 @@ def evaluate(model, val_data, nr_eval, local_rank, writer_val):
         if i == 0 and local_rank == 0:
             for j in range(10):
                 imgs = np.concatenate((merged_img[j], pred[j], gt[j]), 1)[:, :, ::-1]
-                ..add_image(str(j) + '/img', imgs.copy(), nr_eval, dataformats='HWC')
+                writer_val.add_image(str(j) + '/img', imgs.copy(), nr_eval, dataformats='HWC')
                 writer_val.add_image(str(j) + '/flow', flow2rgb(flow0[j][:, :, ::-1]), nr_eval, dataformats='HWC')
     
     eval_time_interval = time.time() - time_stamp