write to excel

isl-org · Dec 14, 2023 · cf9793c · cf9793c
1 parent 1c3e4d0
commit cf9793c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lseg_repri.py b/lseg_repri.py
@@ -307,7 +307,13 @@ def episodic_validate(args):
             with torch.no_grad():
                 for i in range(args.batch_size_val):
                     # load each pair in the episode one by one
-                    batch = next(args.val_loader)
+                    try:
+                        batch = next(args.val_loader)
+                    except:
+                        # for multiple runs
+                        args.val_loader = iter(args.val_loader)
+                        batch = next(args.val_loader)
+
                     qry_img = batch['query_img']
                     q_label = batch['query_mask']
                     spprt_imgs = batch['support_imgs']