Delete sample_config.json with huggingface_hub
Browse files- sample_config.json +0 -48
sample_config.json
DELETED
|
@@ -1,48 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"train_meta": {
|
| 3 |
-
"exp_name": "release",
|
| 4 |
-
"exp_type": "all",
|
| 5 |
-
"condition": true,
|
| 6 |
-
"vae_latent": true,
|
| 7 |
-
"device": "cuda:0",
|
| 8 |
-
"use_dictdataset": true
|
| 9 |
-
},
|
| 10 |
-
|
| 11 |
-
"dependencies": {
|
| 12 |
-
"dataset_path": "./prerequisites/mimic_vae_lite.pt",
|
| 13 |
-
"checkpoints_dir": "./checkpoints",
|
| 14 |
-
"vae_path": "./prerequisites/vae_model.pth",
|
| 15 |
-
"text_embed_path": "./prerequisites/mimic_iv_text_embed.csv"
|
| 16 |
-
},
|
| 17 |
-
|
| 18 |
-
"hyper_para": {
|
| 19 |
-
"epochs": 200,
|
| 20 |
-
"lr": 5e-4,
|
| 21 |
-
"batch_size": 512,
|
| 22 |
-
"num_train_steps": 1000,
|
| 23 |
-
"unet_kernel_size": 7,
|
| 24 |
-
"unet_num_level": 5,
|
| 25 |
-
"beta_start": 0.00085,
|
| 26 |
-
"beta_end": 0.0120
|
| 27 |
-
},
|
| 28 |
-
|
| 29 |
-
"inference_setting": {
|
| 30 |
-
"device": "cuda:0",
|
| 31 |
-
"inference_timestep": 1000,
|
| 32 |
-
"gen_batch": 4,
|
| 33 |
-
"save_img": true,
|
| 34 |
-
"verbose": false,
|
| 35 |
-
"save_path": "./test_sample",
|
| 36 |
-
"vae_path": "./prerequisites/vae_model.pth",
|
| 37 |
-
"unet_path": "./prerequisites/unet.pth",
|
| 38 |
-
"text": "Sinus rhythm.",
|
| 39 |
-
"text_embed_setting": {
|
| 40 |
-
"use_api": false,
|
| 41 |
-
"OPENAI_API_KEY": "",
|
| 42 |
-
"loacal_text_embed_path": "./prerequisites/mimic_iv_text_embed.csv"
|
| 43 |
-
},
|
| 44 |
-
"age": 50,
|
| 45 |
-
"hr": 90,
|
| 46 |
-
"gender": "M"
|
| 47 |
-
}
|
| 48 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|