acc = torch.mean( (torch.sum(lbls == (res > 0.5), 1) == predicted_size).type(torch.float32)) log.append_train(loss, acc) model.eval() N = len(testLoader) for it, (batch, lbls) in enumerate(testLoader): print(str(it) + '/' + str(N)) batch = batch.to(device) lbls = lbls.to(device) res = model(batch) res = torch.sigmoid(res) loss = wce(res, lbls, w_positive_tensor, w_negative_tensor) acc = torch.mean( (torch.sum(lbls == (res > 0.5), 1) == predicted_size).type(torch.float32)) log.append_test(loss, acc) log.save_and_reset() log.plot() scheduler.step()