CUDA_VISIBLE_DEVICES=0 torchrun --standalone --nproc_per_node=gpu cfg/train_semijoint2.py \
	--train_pkl napkin_mnist/synthetic_training_data/synthetic_W1W2XY_300.pkl\
	--val_pkl napkin_mnist/synthetic_training_data/synthetic_W1W2XY_300.pkl\
	--inch 3 \
	--outch 3 \
	--datakey Y \
	--condkey X \
	--labkey_0 W2a \
	--labkey_1 W2b \
	--drop-lab 0 \
	--drop-imgcond 1\
	--moddir napkin_mnist/final_model_NODROP \
	--samdir napkin_mnist/final_model_evals_NODROP \
	--epoch 301 \
	--interval 50
