Inoperative lines of code in the training file

[tmondalvogo]

Hello,
Thank you for sharing the code. I find that there are some inoperative lines of code in the training file. The use of these operations is not clear and it seems that these operations are not needed for training.

KpSFR/train_nn.py

Line 188 in 3d4b796

# TODO: random pick 4

The array ref_v[ ] doesn't have any participation in the training. It is created and then deleted.
The variable is created in line :

KpSFR/train_nn.py

Line 190 in 3d4b796

ref_v = []

Then it is only used to check an assert condition :

KpSFR/train_nn.py

Line 223 in 3d4b796

assert torch.isnan(ref_v).sum() == 0, print('ref_v: ', ref_v)

Then it is deleted :

KpSFR/train_nn.py

Line 273 in 3d4b796

del ref_v

Hence all the operations done
From line:

KpSFR/train_nn.py

Line 189 in 3d4b796

ref_hm = target_dilated_hm.clone()

To line:

KpSFR/train_nn.py

Line 223 in 3d4b796

assert torch.isnan(ref_v).sum() == 0, print('ref_v: ', ref_v)

has no use in the training code i.e. these operations are not used in any sort of loss calculation etc.

Then why they are done in this code? What is the use of it?

To easily follow the code, you may use the cleaner version of the code here :
#4 (comment)