Upload 3 files

config.json

@@ -0,0 +1,33 @@
+{
+    "module": "zea.models.unet",
+    "class_name": "UNet",
+    "config": {
+        "name": "unet",
+        "trainable": true,
+        "dtype": {
+            "module": "keras",
+            "class_name": "DTypePolicy",
+            "config": {
+                "name": "float32"
+            },
+            "registered_name": null
+        },
+        "input_shape": [
+            128,
+            128,
+            1
+        ],
+        "input_range": [
+            -1,
+            1
+        ],
+        "widths": [
+            32,
+            64,
+            96,
+            128
+        ],
+        "block_depth": 2
+    },
+    "registered_name": "UNet"
+}

model.weights.h5

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

train_config.yaml

@@ -0,0 +1,69 @@
+seed: 42
+data:
+  image_shape:
+  - 128
+  - 128
+  - 1
+  n_frames: 1
+  dataset_repetitions: 1
+  batch_size: 64
+  train_folder: echonet/train
+  val_folder: echonet/val
+  num_img: null
+  extension: hdf5
+  hdf5_key: data/image
+  image_range:
+  - -60
+  - 0
+  normalization_range:
+  - -1
+  - 1
+  resize_type: resize
+  local: false
+  shuffle: true
+  usbmd_config: configs/usbmd/config_echonet.yaml
+  overlapping_blocks: true
+  augmentation:
+    training:
+    - random_flip:
+        mode: horizontal
+    validation: null
+
+evaluation:
+  batch_size: 16
+model:
+  name: unet
+  widths:
+  - 32
+  - 64
+  - 96
+  - 128
+  block_depth: 2
+  run_eagerly: false
+  image_shape:
+  - 128
+  - 128
+  - 1
+  image_range:
+  - -1
+  - 1
+optimization:
+  num_epochs: 100
+  ema: 0.999
+  learning_rate: 0.0001
+  weight_decay: 0.0001
+  loss: mse
+  steps_per_epoch: 2000
+device: gpu:0
+hide_devices: null
+corruption_operator:
+  type: random_line_mask
+  kwargs:
+    mask_factor: 0.75
+    line_width: 1
+    image_shape:
+    - 128
+    - 128
+    - 1
+    remove_lines: false
+wandb: true