数据集训练问题

[star-cs]

作者你好，请问可以提供LA，PROMISE两个数据集的Dataset和train的细节代码吗，不胜感激。

[Cccccczh404]

您好，以下LA和PROMISE两个数据集的training scripts
CUDA_VISIBLE_DEVICES="0" python train.py --root_path /PROMISE/train_npz --output ./results/PROMISE_3/ --split='train' --batch_size=8 --base_lr=0.0026 --img_size=512 --warmup --dice_param=0.9 --AdamW --max_epochs=250 --stop_epoch=250 --num_classes=1 --dataset='PROMISE' --list_dir='./lists/lists_PROMISE_3' --ckpt='checkpoints/sam_vit_b_01ec64.pth'
CUDA_VISIBLE_DEVICES="0" python train.py --root_path /LA/train_npz --output ./results/LA_4/ --split='train' --batch_size=8 --base_lr=0.0028 --img_size=512 --warmup --dice_param=0.9 --AdamW --max_epochs=250 --stop_epoch=250 --num_classes=1 --dataset='LA' --list_dir='./lists/lists_LA_4' --ckpt='checkpoints/sam_vit_b_01ec64.pth'