switch to torchmetrics (#590)

* switch to torchmetrics

* 0.2.0

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>

pl_bolts/callbacks/ssl_online.py

@@ -2,10 +2,10 @@
 
 import torch
 from pytorch_lightning import Callback, LightningModule, Trainer
-from pytorch_lightning.metrics.functional import accuracy
 from torch import device, Tensor
 from torch.nn import functional as F
 from torch.optim import Optimizer
+from torchmetrics.functional import accuracy
 
 
 class SSLOnlineEvaluator(Callback):  # pragma: no cover

pl_bolts/models/regression/logistic_regression.py

@@ -2,12 +2,12 @@
 
 import pytorch_lightning as pl
 import torch
-from pytorch_lightning.metrics.functional import accuracy
 from torch import nn
 from torch.nn import functional as F
 from torch.nn.functional import softmax
 from torch.optim import Adam
 from torch.optim.optimizer import Optimizer
+from torchmetrics.functional import accuracy
 
 
 class LogisticRegression(pl.LightningModule):

pl_bolts/models/self_supervised/ssl_finetuner.py

@@ -2,8 +2,8 @@
 
 import pytorch_lightning as pl
 import torch
-from pytorch_lightning.metrics import Accuracy
 from torch.nn import functional as F
+from torchmetrics import Accuracy
 
 from pl_bolts.models.self_supervised import SSLEvaluator
 

requirements.txt

@@ -1,2 +1,3 @@
 torch>=1.6
+torchmetrics>=0.2.0
 pytorch-lightning>=1.1.1