CUDA_VISIBLE_DEVICE=0 python federated_main_ibpcnn_v3.py --model=ibpcnn --dataset=cifar --gpu=0 --iid=0 --epochs=20 --local_ep=5 --truncation=60 --conv_truncation=10 --a_prior=50 --lr=0.02 --klr=0.1 --klb=0.1 --klv=0.1 --num_channels=3
