Merge pull request #9 from PINTO0309/pinto0309work

Avoiding "local variable'i_val' referenced before assignment"
This commit is contained in:
Xuebin Qin 2022-07-30 09:27:58 -07:00 committed by GitHub
commit bdf1a300d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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']