search

cvlab-stonybrook / SelfMedMAE

Code for ISBI 2023 paper "Self Pre-training with Masked Autoencoders for Medical Image Classification and Segmentation"
Apache License 2.0
112 stars 12 forks source link

library versions #4

Closed yuankaiWang closed 10 months ago

yuankaiWang commented 1 year ago

could anyone who share their version of library in this project. Much appraciate

jackhu-bme commented 1 year ago

torch torchvision torchaudio omegaconf timm=0.6.13 scikit-learn monai==1.0.0 matplotlib torchprofile nibabel

jackhu-bme commented 1 year ago

you can install the latest versions of other libs, but monai and timm can not be too high

BannyStone commented 10 months ago

Thanks @jackhu-bme ! I just updated the info in README. Here are the package versions I used: pytorch=1.7.1=py3.8_cuda10.1.243_cudnn7.6.3_0 torchvision=0.8.2=py38_cu101 monai-weekly==0.9.dev2152 nibabel==3.2.1 omegaconf==2.1.1 timm==0.4.12