diff --git a/EchoNet/main.sh b/EchoNet/main.sh index 81dace1..af67f21 100755 --- 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 \ No newline at end of file +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 index 297e2c2..4873096 100644 --- 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 ``` \ No newline at end of file diff --git a/src/config.py b/src/config.py index e448e5a..f25e233 100644 --- 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 }, -} \ No newline at end of file +}