# With All DA / not OC :
#python -m ipdb -c c train_wandb.py --add_descriptive_test=True --add_discriminative_test=True --agent_loss_type=Hinge --agent_nbr_latent_dim=32 --arch=BN+BetaVAEEncoderOnly3x3 --baseline_only=False --batch_size=64 --dataset=3dshapes --descriptive=True --descriptive_ratio=0 --dis_metric_resampling=True --distractor_sampling=uniform --dropout_prob=0 --emb_dropout_prob=0 --epoch=21 --graphtype=obverter --lr=0.0001 --max_sentence_length=10 --metric_active_factors_only=True --metric_batch_size=64 --metric_epoch_period=10 --metric_resampling=True --mini_batch_size=64 --nb_3dshapespybullet_colors=10 --nb_3dshapespybullet_samples=10 --nb_3dshapespybullet_shapes=10 --nb_3dshapespybullet_train_colors=5 --nbr_discriminative_test_distractors=7 --nbr_distractors=3 --nbr_eval_points=5000 --nbr_train_points=5000 --object_centric=False --obverter_nbr_games_per_round=2 --obverter_nbr_head_outputs=2 --obverter_sampling=False --obverter_sampling_round_alternation_only=True --obverter_threshold_to_stop_message_generation=0.95 --obverter_use_decision_head=True --parallel_TS_worker=32 --parent_folder=./3dshapesXL_obverter_21epochs --resizeDim=64 --seed=10 --shared_architecture=False --symbol_embedding_size=64 --symbol_processing_nbr_hidden_units=128 --synthetic_progression_end=10 --train_test_split_strategy=combinatorial2-FloorHue-1-S5-WallHue-1-S5-ObjectHue-1-S5-Scale-8-N-Shape-1-N-Orientation-1-N --use_cuda=True --vae_factor_gamma=0 --vae_gaussian=False --vae_lambda=1 --vae_nbr_latent_dim=32 --visual_context_consistent_obverter=False --vocab_size=10 --with_BN_in_obverter_decision_head=False --with_DP_in_obverter_decision_head=False --with_baseline=False --with_gaussian_blur_augmentation=True --with_color_jitter_augmentation=True --egocentric=True

# With all DA / not OC + DP :
python -m ipdb -c c train_wandb.py --add_descriptive_test=True --add_discriminative_test=True --agent_loss_type=Hinge --agent_nbr_latent_dim=32 --arch=BN+BetaVAEEncoderOnly3x3 --baseline_only=False --batch_size=64 --dataset=3dshapes --descriptive=True --descriptive_ratio=0 --dis_metric_resampling=True --distractor_sampling=uniform --dropout_prob=0 --emb_dropout_prob=0 --epoch=21 --graphtype=obverter --lr=0.0001 --max_sentence_length=10 --metric_active_factors_only=True --metric_batch_size=64 --metric_epoch_period=10 --metric_resampling=True --mini_batch_size=64 --nb_3dshapespybullet_colors=10 --nb_3dshapespybullet_samples=10 --nb_3dshapespybullet_shapes=10 --nb_3dshapespybullet_train_colors=5 --nbr_discriminative_test_distractors=7 --nbr_distractors=3 --nbr_eval_points=5000 --nbr_train_points=5000 --object_centric=False --obverter_nbr_games_per_round=2 --obverter_nbr_head_outputs=2 --obverter_sampling=False --obverter_sampling_round_alternation_only=True --obverter_threshold_to_stop_message_generation=0.95 --obverter_use_decision_head=True --parallel_TS_worker=32 --parent_folder=./3dshapesXL_obverter_21epochs --resizeDim=64 --seed=10 --shared_architecture=False --symbol_embedding_size=64 --symbol_processing_nbr_hidden_units=128 --synthetic_progression_end=10 --train_test_split_strategy=combinatorial2-FloorHue-1-S5-WallHue-1-S5-ObjectHue-1-S5-Scale-8-N-Shape-1-N-Orientation-1-N --use_cuda=True --vae_factor_gamma=0 --vae_gaussian=False --vae_lambda=1 --vae_nbr_latent_dim=32 --visual_context_consistent_obverter=False --vocab_size=10 --with_BN_in_obverter_decision_head=False --with_DP_in_obverter_decision_head=False --with_baseline=False --with_gaussian_blur_augmentation=True --with_color_jitter_augmentation=True --egocentric=True --DP_in_obverter_decision_head=0.5

