From ba3a1ae7898b31002dd111ec5afb435c4f96aece Mon Sep 17 00:00:00 2001
From: David de la Iglesia <daviddelaiglesiacastro@gmail.com>
Date: Thu, 1 Feb 2024 12:27:30 +0000
Subject: [PATCH] Tune train.arch

---
 dvc.lock    | 30 +++++++++++++++---------------
 dvc.yaml    |  2 +-
 params.yaml |  9 ++++-----
 3 files changed, 20 insertions(+), 21 deletions(-)

diff --git a/dvc.lock b/dvc.lock
index 98ab587e..ddd73620 100644
--- a/dvc.lock
+++ b/dvc.lock
@@ -48,7 +48,7 @@ stages:
           random_seed: 42
         train:
           valid_pct: 0.1
-          arch: shufflenet_v2_x2_0
+          arch: resnet34
           img_size: 256
           batch_size: 8
           fine_tune_args:
@@ -57,16 +57,16 @@ stages:
     outs:
     - path: models/model.pkl
       hash: md5
-      md5: 63cb30df484bfa1d6ea2cdebac74876c
-      size: 201725
+      md5: 3f9fb1dbc04046ddb6f6a96ac967c606
+      size: 207334766
     - path: models/model.pth
       hash: md5
-      md5: 07b113fe1ab01de2d8a5453dccacdd3e
-      size: 165147
+      md5: df30344c2f629c248855169e5f525fc0
+      size: 207285126
     - path: results/train
       hash: md5
-      md5: a26429ff680d9b01c7e92043821bf41c.dir
-      size: 955
+      md5: 8d42ff1cae8857bf1a35f7f3ccee041a.dir
+      size: 957
       nfiles: 5
   evaluate:
     cmd: python src/evaluate.py
@@ -78,8 +78,8 @@ stages:
       nfiles: 24
     - path: models/model.pkl
       hash: md5
-      md5: 63cb30df484bfa1d6ea2cdebac74876c
-      size: 201725
+      md5: 3f9fb1dbc04046ddb6f6a96ac967c606
+      size: 207334766
     - path: src/evaluate.py
       hash: md5
       md5: 84d2fd3b371546730396a763a51527a0
@@ -93,8 +93,8 @@ stages:
     outs:
     - path: results/evaluate
       hash: md5
-      md5: 34985c391291e22ac0cfdf6e83ec7268.dir
-      size: 1257936
+      md5: ae80d36acde3d28b13817c4953a7a94a.dir
+      size: 1254205
       nfiles: 11
   sagemaker:
     cmd: cp models/model.pth sagemaker/code/model.pth && cd sagemaker && tar -cpzf
@@ -102,10 +102,10 @@ stages:
     deps:
     - path: models/model.pth
       hash: md5
-      md5: 07b113fe1ab01de2d8a5453dccacdd3e
-      size: 165147
+      md5: df30344c2f629c248855169e5f525fc0
+      size: 207285126
     outs:
     - path: model.tar.gz
       hash: md5
-      md5: c615b812da343d71fcda080597a24525
-      size: 145778
+      md5: 4ba1d6aa076c3ca720aeac166606aa66
+      size: 180319319
diff --git a/dvc.yaml b/dvc.yaml
index f89da695..ad0a3938 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -17,7 +17,7 @@ artifacts:
     - cv
     - segmentation
     - satellite-images
-    - shufflenet_v2_x2_0
+    - resnet34
 stages:
   data_split:
     cmd: python src/data_split.py
diff --git a/params.yaml b/params.yaml
index 767fc214..cbe13fae 100644
--- a/params.yaml
+++ b/params.yaml
@@ -2,17 +2,16 @@ base:
   random_seed: 42
 
 data_split:
-  test_regions: 
-    - REGION_1
-
+  test_regions:
+  - REGION_1
 train:
   valid_pct: 0.1
-  arch: shufflenet_v2_x2_0
+  arch: resnet34
   img_size: 256
   batch_size: 8
   fine_tune_args:
     epochs: 8
-    base_lr: 0.01
+    base_lr: 0.01
 
 evaluate:
   n_samples_to_save: 10