minor change

kdhht2334 · Sep 20, 2022 · 026fa2c · 026fa2c
1 parent a48f6bf
commit 026fa2c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -63,6 +63,10 @@ pip install -r requirements.txt
 - [x] Upload train/eval files
 
 
+### Note
+- In case of Mlp-Mixer, please refer official repository [(link)](https://github.com/google-research/vision_transformer#mlp-mixer)
+
+
 
 ### Citation
 

diff --git a/run.sh b/run.sh
@@ -4,7 +4,7 @@ CUDA_VISIBLE_DEVICES=1 python main.py \
 	--model alexnet \
 	--sinkhorn 1 \
 	--relevance_weighting 1 \
-	--weight_sampling gumbel \
+	--weight_sampling gumbel-softmax \
 	--lr 5e-5 \
 	--no_domain 5 \
 	--topk 5 \