forked from zhangxiaoyu11/OmiEmbed
-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathargs.json
1 lines (1 loc) · 1.2 KB
/
args.json
1
{"gpu_ids": [0], "checkpoints_dir": "./checkpoints", "experiment_name": "visual_test", "omics_mode": "abc", "data_root": "./data/ABC_inter", "batch_size": 32, "num_threads": 0, "set_pin_memory": false, "not_stratified": false, "use_sample_list": true, "use_feature_lists": false, "detect_na": false, "file_format": "npy", "model": "vae_classifier", "net_VAE": "fc_sep", "net_down": "multi_FC_classifier", "norm_type": "batch", "filter_num": 8, "conv_k_size": 9, "dropout_p": 0.2, "leaky_slope": 0.2, "latent_space_dim": 128, "seed": 42, "init_type": "normal", "init_gain": 0.02, "recon_loss": "BCE", "reduction": "mean", "k_kl": 0.01, "k_embed": 0.001, "deterministic": false, "detail": false, "epoch_to_load": "latest", "experiment_to_load": "test", "epoch_num_p1": 50, "epoch_num_p2": 0, "epoch_num_p3": 0, "lr": 0.0001, "beta1": 0.5, "lr_policy": "linear", "epoch_count": 1, "epoch_num_decay": 50, "decay_step_size": 50, "weight_decay": 0.0001, "continue_train": false, "save_model": true, "save_epoch_freq": -1, "print_freq": 1, "train_ratio": 0.8, "test_ratio": 0.2, "class_num": 0, "isTrain": true, "isTest": true, "epoch_num": 50, "downstream_task": "classification", "add_channel": false, "ch_separate": true, "omics_num": 3}