diff --git a/requirements.txt b/requirements.txt index 60821c49703d2a03c91827065cbdf11995db477e..580db03825fae26b8e59b485f4502b0b3e345e8c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,5 @@ pandas~=2.2.0 numpy~=1.26.2 matplotlib~=3.8.2 wandb~=0.16.2 -torchmetrics~=1.3.0.post0 \ No newline at end of file +torchmetrics~=1.3.0.post0 +torcheval~=0.0.7 \ No newline at end of file