resnet50

sunbing7 · Jun 12, 2023 · 839d205 · 839d205
1 parent dd9509e
commit 839d205
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src_torch/main.py b/src_torch/main.py
@@ -105,6 +105,12 @@ def moth():
             epsilon, k, a = 0.1, 8, 0.02
         elif args.dataset == 'fmnist':
             epsilon, k, a = 8/255, 7, 2/255
+        elif args.dataset == 'mnistm':
+            epsilon, k, a = 8/255, 7, 2/255
+        elif args.dataset == 'asl':
+            epsilon, k, a = 8/255, 7, 2/255
+        elif args.dataset == 'caltech':
+            epsilon, k, a = 8/255, 7, 2/255
 
     # hardening iterations
     max_warmup_steps = warmup_steps * num_classes