mmnist_MMVAEconfig2_seed_1_ratio_0_c / training_config.json
asenella's picture
Upload 6 files
a225dd3
{"name": "BaseTrainerConfig", "output_dir": "compare_on_mmnist/config2/MMVAE/seed_1/missing_ratio_0/", "per_device_train_batch_size": 256, "per_device_eval_batch_size": 64, "num_epochs": 100, "train_dataloader_num_workers": 0, "eval_dataloader_num_workers": 0, "optimizer_cls": "Adam", "optimizer_params": {}, "scheduler_cls": "ReduceLROnPlateau", "scheduler_params": {"patience": 30}, "learning_rate": 0.001, "steps_saving": null, "steps_predict": 5, "keep_best_on_train": false, "seed": 1, "no_cuda": false, "world_size": -1, "local_rank": -1, "rank": -1, "dist_backend": "nccl", "master_addr": "localhost", "master_port": "12345", "start_keep_best_epoch": 0, "drop_last": false}