From 9ff8e57a5f93381772d2d6373ec5e08a4a6a6307 Mon Sep 17 00:00:00 2001 From: TufailRouf <121852083+TufailRouf@users.noreply.github.com> Date: Wed, 1 Feb 2023 08:42:56 +0000 Subject: [PATCH] Update run_vanilla_kd.sh --- code/image_classfication/run_vanilla_kd.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/image_classfication/run_vanilla_kd.sh b/code/image_classfication/run_vanilla_kd.sh index 3c402b8..1dafa3a 100755 --- a/code/image_classfication/run_vanilla_kd.sh +++ b/code/image_classfication/run_vanilla_kd.sh @@ -5,7 +5,7 @@ do python main.py --device=cuda:0 \ --teacher=wrn-28-4 \ --student=wrn-28-4 \ - --dataset=cifar100 \ + --dataset=cifar10 \ --batch_size=128 \ --num_epochs=200 \ --mode=crop \