actu-magnitude-regression / requirements.txt
DarthReca's picture
Create requirements.txt
c85c48f verified
transformers
torch
timm
segmentation-models-pytorch
einops
numpy