Requirements: Python 3.12.3, CUDA 12.4 (for GPU), Linux (Ubuntu 20.04+ recommended), conda & pip.
Steps:
-
Create Conda env:
conda create -n mdpr_project python=3.12.3 && conda activate mdpr_project -
Install deps:
pip install -r requirements.txt -
Verify:
python -c "import torch; print(torch.__version__, torch.cuda.is_available())"(expect2.5.1+cu124 True)
Run:
-
Prepare dataset in
./data(e.g.,data/train,data/val). -
Run:
python main_maple.py --dataset cifar100 --batch_size 128 --epochs 20 --lr 0.001 -
Outputs: Model weights in
./checkpoints/cifar100/(e.g.,best_best.pth.tar), logs and test results in./checkpoints/cifar100/(e.g.,mdpr_clip_maple_log.txt,mdpr_clip_maple_best.txt).