Closed Dylan-H-Wang closed 3 years ago
Hi, In the test() function of test.py, is that a typo when you are writing hr which actually should be bpm?
test()
test.py
hr
bpm
def test(): net.eval() test_loss = 0; for (data, bpm, fps, bvp, idx) in test_loader: data = Variable(data); hr = Variable(hr.view(-1,1)); data, hr = data.cuda(), hr.cuda(); feat_hr, feat_n, output, img_out, feat_hrf1, feat_nf1, hrf1, idx1, feat_hrf2, feat_nf2, hrf2, idx2, ecg, ecg1, ecg2 = net(data, epoch); loss = lossfunc_HR(output, hr); test_loss += loss.item();
Thank you !! The typo has been corrected.
Hi, In the
test()
function oftest.py
, is that a typo when you are writinghr
which actually should bebpm
?