Genentech / gReLU

gReLU is a python library to train, interpret, and apply deep learning models to DNA sequences.
https://genentech.github.io/gReLU/
MIT License
230 stars 23 forks source link

Pin torchmetrics version #51

Closed gokceneraslan closed 1 month ago

gokceneraslan commented 1 month ago

to fix the following error:

model = grelu.resources.load_model(project="borzoi",model_name="human_fold0", device='cuda:0')

File ~/mambaforge/envs/scverse/lib/python3.11/site-packages/grelu/lightning/metrics.py:137, in PearsonCorrCoef.__init__(self, num_outputs, average)
    135 def __init__(self, num_outputs: int = 1, average: bool = True) -> None:
    136     super().__init__()
--> 137     self.pearson = torchmetrics.PearsonCorrCoef(num_outputs=num_outputs)
    138     self.average = average

TypeError: PearsonCorrCoef.__init__() got an unexpected keyword argument 'num_outputs'