Upload files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+mimic_iv_text_embed.csv filter=lfs diff=lfs merge=lfs -text

mimic_iv_text_embed.csv

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:113eccc685c97aa8c48b837f3195a35e3ba3a4b549fd7103829a18537c9c2bff
+size 107145253

mimic_vae.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa030135d76bf5c277001aba1208fe8d5f81e5c9e8409e3d11ac4ca85dbbda7f
+size 2170256846

mimic_vae_lite.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:17f9352f761dadea53d8f91da36bf7c34cfcb75749441b98252b9a45caff0095
+size 39692196

sample_config.json

@@ -0,0 +1,48 @@
+{ 
+    "train_meta": {
+        "exp_name": "release", 
+        "exp_type": "all", 
+        "condition": true, 
+        "vae_latent": true, 
+        "device": "cuda:0", 
+        "use_dictdataset": true 
+    }, 
+
+    "dependencies": {
+        "dataset_path": "./prerequisites/mimic_vae_lite.pt",
+        "checkpoints_dir": "./checkpoints", 
+        "vae_path": "./prerequisites/vae_model.pth", 
+        "text_embed_path": "./prerequisites/mimic_iv_text_embed.csv"
+    },
+
+    "hyper_para": {
+        "epochs": 200, 
+        "lr": 5e-4, 
+        "batch_size": 512, 
+        "num_train_steps": 1000, 
+        "unet_kernel_size": 7, 
+        "unet_num_level": 5, 
+        "beta_start": 0.00085, 
+        "beta_end": 0.0120
+    }, 
+
+    "inference_setting": {
+        "device": "cuda:0",
+        "inference_timestep": 1000, 
+        "gen_batch": 4, 
+        "save_img": true, 
+        "verbose": false,
+        "save_path": "./test_sample", 
+        "vae_path": "./prerequisites/vae_model.pth", 
+        "unet_path": "./prerequisites/unet.pth", 
+        "text": "Sinus rhythm.", 
+        "text_embed_setting": {
+            "use_api": false, 
+            "OPENAI_API_KEY": "", 
+            "loacal_text_embed_path": "./prerequisites/mimic_iv_text_embed.csv"
+        },
+        "age": 50,
+        "hr": 90, 
+        "gender": "M"
+    }
+}

unet.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b96b7a975d32e5927193c1a3153e673b2d109cc3ae421d04cbd19f9d64d0c6f
+size 5871546

vae_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2b4ef8060e412bd503f46b66dd44fc460e67a89de0ae1809dcc771358b3fa93
+size 118289354