在 pytorch 中使用超过 1 个指标

Question

我对 Tenserflow 有一些经验，但我是 pytorch 的新手。有时我需要不止 1 个指标来检查训练的准确性。在 Tenserflow 中，我以前是这样做的，如下所示。但我想知道如何在 pytorch 中列出超过 1 个指标。

LR = 0.0001
optim = keras.optimizers.Adam(LR)

dice_loss_se2 = sm.losses.DiceLoss()
mae = tf.keras.losses.MeanAbsoluteError( )
metrics = [ mae,sm.metrics.IOUScore(threshold=0.5), sm.metrics.FScore(threshold=0.5) , dice_loss_se2]

model.compile(optimizer=optim,loss= dice_loss_se2,metrics= metrics)

Answer 1

在 pytorch 中，训练主要是通过循环完成的，因此您可以通过每个步骤进行定义，您可以使用 torchmetrics 等包运行每个指标都是一个示例：

import torchmetrics

for step, (test_image, test_labels) in tqdm(enumerate(test_dataloader), total=len(test_dataloader)):
        test_batch_image = test_image.to('cuda')
        test_batch_label = test_labels.to('cuda')
        targets.append(test_labels)
        
        with torch.no_grad():
            logits = model(test_batch_image)
        
        loss = criterion(logits, test_batch_label)
        test_loss += loss.item()
        
        preds.append(logits.detach().cpu().numpy().argmax(axis=1))
    
    preds = torch.tensor(np.concatenate(preds))
    targets = torch.tensor(np.concatenate(targets))
    print('[Epoch %d] Test loss: %.3f' %(epoch + 1, test_loss/ len(test_dataloader)))
    print('Accuracy: {}%'.format(round(torchmetrics.functional.accuracy(target=targets, preds=preds).item() * 100), 2))

在 pytorch 中使用超过 1 个指标

Using more than 1 metric in pytorch

python

keras

tensorflow

pytorch