hushon / noisy-label

12 stars 0 forks source link

noisy-label

Setting up

docker pull hushon/pytorch:2.0

Run training

CUDA_VISIBLE_DEVICES=0 python train.py ./configs/train_base.yaml