update examples

DSGT-DLP · Jun 24, 2023 · 0ba779f · 0ba779f
1 parent 3cb00b5
commit 0ba779f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/openapi/paths/train/img-run.yaml b/openapi/paths/train/img-run.yaml
@@ -87,17 +87,24 @@ post:
             parameters_data:
               problem_type: "classification"
               criterion: "CELOSS"
-              optimizer_name: "SGD"
+              optimizer_name: "ADAM"
               shuffle: true
               epochs: 10
               test_size: 0.2
               batch_size: 20
               layers:
                 [
-                  "nn.Linear(4, 10)",
-                  "nn.ReLU()",
-                  "nn.Linear(10, 3)",
-                  "nn.Softmax()",
+                  "nn.Conv2d(1, 5, 3, 1, 1)",
+                  "nn.MaxPool2d(3, 1)",
+                  "nn.Flatten()",
+                  "nn.Linear(500, 10)",
+                  "nn.Sigmoid()",
+                ]
+              train_transform:
+                [
+                  "torchvision.transforms.Grayscale()",
+                  "torchvision.transforms.toTensor()",
+                  "torchvision.transforms.Resize((32, 32))",
                 ]
     required: true
   responses:

diff --git a/openapi/paths/train/tabular-run.yaml b/openapi/paths/train/tabular-run.yaml
@@ -98,10 +98,10 @@ post:
               batch_size: 20
               layers:
                 [
-                  "nn.Linear(4, 10)",
-                  "nn.ReLU()",
                   "nn.Linear(10, 3)",
-                  "nn.Softmax()",
+                  "nn.ReLU()",
+                  "nn.Linear(3, 10)",
+                  "nn.Softmax(-1)",
                 ]
     required: true
   responses: