CUDA_VISIBLE_DEVICES=0,1 torchrun --standalone --nproc_per_node=gpu cfg/train_joint2.py \
	--train_pkl napkin_mnist/base_data/napkin_mnist_train.pkl \
	--val_pkl napkin_mnist/base_data/napkin_mnist_val.pkl \
	--datakey X,Y \
	--condkey W1 \
	--labkey_0 W2a \
	--clsnum_0 10 \
	--labkey_1 W2b \
	--clsnum_1 2 \
	--moddir napkin_mnist/synthetic_model \
	--samdir napkin_mnist/synthetic_model_evals \
	--interval 50 \
	--epoch 1001 \
	--w 1.0