diff --git a/IS-Net/train_valid_inference_main.py b/IS-Net/train_valid_inference_main.py index 209567d..4448809 100644 --- a/IS-Net/train_valid_inference_main.py +++ b/IS-Net/train_valid_inference_main.py @@ -200,6 +200,8 @@ def valid_gt_encoder(net, valid_dataloaders, valid_datasets, hypar, epoch=0): MAE = np.zeros((val_num)) val_cnt = 0.0 + i_val = None + for i_val, data_val in enumerate(valid_dataloader): # imidx_val, inputs_val, labels_val, shapes_val = data_val['imidx'], data_val['image'], data_val['label'], data_val['shape']