CUDA_VISIBLE_DEVICES=0,1 torchrun --standalone --nproc_per_node=gpu cfg/train_imgcond.py \
	--train_pkl napkin_mnist/base_data/napkin_mnist_train.pkl \
	--val_pkl napkin_mnist/base_data/napkin_mnist_val.pkl \
	--datakey Y \
	--condkey X \
	--moddir napkin_mnist/conditional_model_Y_X \
	--samdir napkin_mnist/conditional_model_Y_X_evals \
	--interval 50 \
	--epoch 1001 \
	--w 1.0