update Echo

huaxiuyao · Nov 15, 2022 · ad01b60 · ad01b60
1 parent bae8087
commit ad01b60
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/EchoNet/main.sh b/EchoNet/main.sh
@@ -9,4 +9,4 @@ echonet video --batch_size 10 --device cuda --num_workers 0 --num_epochs 20 --mi
 echonet video --batch_size 10 --device cuda --num_workers 0 --num_epochs 20 --mixtype random --run_test True
 
 # kde
-echonet video --batch_size 10 --device cuda --num_workers 0 --num_epochs 20 --mixtype kde --bandwidth 100.0 --run_test True
+echonet video --batch_size 10 --device cuda --num_workers 0 --num_epochs 20 --mixtype kde --bandwidth 50.0 --run_test True
diff --git a/README.md b/README.md
@@ -137,5 +137,5 @@ python echonet/__main__.py segmentation --save_video
 
 The command to run C-Mixup on EchoNet is:
 ```
-echonet video --batch_size 10 --device cuda --num_workers 0 --num_epochs 20 --mixtype kde --bandwidth 100.0 --run_test True
+echonet video --batch_size 10 --device cuda --num_workers 0 --num_epochs 20 --mixtype kde --bandwidth 50.0 --run_test True
 ```
diff --git a/src/config.py b/src/config.py
@@ -91,7 +91,7 @@
         'output_fun':'Linear',
     },
 
-    'RotateFashionMNIST':{
+    'RCF_MNIST':{
         'batch_size': 64,
         'num_epochs': 30,
         'optimiser_args': {
@@ -175,4 +175,4 @@
         'holdout_fraction': 0.2,
         'uda_holdout_fraction': 0., # For domain adaptation, % of test to use unlabeled for training
     },
-}
+}