Laiyf commited on
Commit
cebc904
·
verified ·
1 Parent(s): 0bfaf24

Delete sample_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- }