search

ecm200 / caltech_birds

A set of notebooks as a guide to the process of fine-grained image classification of birds species, using PyTorch based deep neural networks.
41 stars 7 forks source link

Integrate torchmetrics package for all metrics used. #18

Open ecm200 opened 3 years ago

ecm200 commented 3 years ago

Use the standalone torchmetrics package for all metrics usage across the functions.

ecm200 commented 3 years ago

Use package TorchMetrics for all metrics.