Update yaml configs

configs/data.yaml/cifar10.test

@@ -1,7 +0,0 @@
-class_or_func: CIFAR10
-module_path: torchvision.datasets
-args: []
-kwargs:
-  train: False
-  download: True
-  transform: null

configs/data.yaml/cifar10.train

@@ -1,7 +0,0 @@
-class_or_func: CIFAR10
-module_path: torchvision.datasets
-args: []
-kwargs:
-  train: True
-  download: True
-  transform: null

configs/yaml.data/cifar10.test

@@ -0,0 +1,22 @@
+class_or_func: CIFAR10
+module_path: torchvision.datasets
+args: []
+kwargs:
+  train: False
+  download: True
+  transform:
+    class_or_func: Compose
+    module_path: torchvision.transforms
+    args:
+      -
+        - class_or_func: ToTensor
+          module_path: torchvision.transforms
+          args: []
+          kwargs: {}
+        - class_or_func: Normalize
+          module_path: torchvision.transforms
+          args: []
+          kwargs:
+            mean: (0.491, 0.482, 0.447)
+            std: (0.247, 0.244, 0.262)
+    kwargs: {}

configs/yaml.data/cifar10.train

@@ -0,0 +1,30 @@
+class_or_func: CIFAR10
+module_path: torchvision.datasets
+args: []
+kwargs:
+  train: True
+  download: True
+  transform:
+    class_or_func: Compose
+    module_path: torchvision.transforms
+    args:
+      -
+        - class_or_func: RandomHorizontalFlip
+          module_path: torchvision.transforms
+          args: []
+          kwargs: {}
+        - class_or_func: RandomCrop
+          module_path: torchvision.transforms
+          args: [32]
+          kwargs: {padding: 4}
+        - class_or_func: ToTensor
+          module_path: torchvision.transforms
+          args: []
+          kwargs: {}
+        - class_or_func: Normalize
+          module_path: torchvision.transforms
+          args: []
+          kwargs:
+            mean: (0.491, 0.482, 0.447)
+            std: (0.247, 0.244, 0.262)
+    kwargs: {}

configs/yaml.loss/cross-entropy

@@ -0,0 +1,4 @@
+class_or_func: CrossEntropyLoss
+module_path: torch.nn
+args: []
+kwargs: {}

configs/yaml.model/vit-cifar10.s0

@@ -0,0 +1,4 @@
+class_or_func: get_transformer
+module_path: xautodl.xmodels.transformers
+args: [vit-cifar10-p4-d4-h4-c32]
+kwargs: {}

configs/yaml.opt/vit.cifar

@@ -0,0 +1,7 @@
+class_or_func: Adam
+module_path: torch.optim
+args: []
+kwargs:
+  betas: [0.9, 0.999]
+  weight_decay: 0.1
+  amsgrad: False