🔥 Change tests optimizer, workflow dependencies

kaylode · Nov 4, 2023 · c92f407 · c92f407 · github-actions · Nov 4, 2023
1 parent 15ccc53
commit c92f407
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 8 deletions.
diff --git a/.github/workflows/clf.yml b/.github/workflows/clf.yml
@@ -3,6 +3,7 @@ on:
   pull_request:
     branches:
       - master
+      - dev
   push:
     branches:
       - master

diff --git a/.github/workflows/segm.yml b/.github/workflows/segm.yml
@@ -3,6 +3,7 @@ on:
   pull_request:
     branches:
       - master
+      - dev
   push:
     branches:
       - master

diff --git a/.github/workflows/tablr.yml b/.github/workflows/tablr.yml
@@ -3,6 +3,7 @@ on:
   pull_request:
     branches:
       - master
+      - dev
   push:
     branches:
       - master
@@ -32,7 +33,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
-          pip install -e '.[dev,tabular,tabular_classification]'
+          pip install -e '.[dev,ml]'
           pip install dvc dvc-gdrive
           npm install -g markdown-table-cli
       - name: Download data

diff --git a/tests/classification/configs/base/optimizer.yaml b/tests/classification/configs/base/optimizer.yaml
@@ -9,6 +9,8 @@ optimizer:
 scheduler:
   name: SchedulerWrapper
   args:
-    scheduler_name: tf_cosinewarmup
-    num_warmup_steps: 100
-    num_cycles: 0.5
+    scheduler_name: cosine2
+    t_initial: 7
+    t_mul: 0.9
+    eta_mul: 0.9
+    eta_min: 1.0e-06
diff --git a/tests/semantic/configs/base/optimizer.yaml b/tests/semantic/configs/base/optimizer.yaml
@@ -9,6 +9,8 @@ optimizer:
 scheduler:
   name: SchedulerWrapper
   args:
-    scheduler_name: tf_cosinewarmup
-    num_warmup_steps: 100
-    num_cycles: 0.5
+    scheduler_name: cosine2
+    t_initial: 7
+    t_mul: 0.9
+    eta_mul: 0.9
+    eta_min: 1.0e-06
diff --git a/theseus/base/callbacks/__init__.py b/theseus/base/callbacks/__init__.py
@@ -5,7 +5,6 @@
     RichModelSummary,
 )
 
-# from theseus.ml.callbacks.base_callbacks import Callbacks, CallbacksList
 from theseus.registry import Registry
 
 from .checkpoint_callback import TorchCheckpointCallback
number	value	datetime_start	datetime_complete	duration	params_optimizer.args.lr	best_key	model_name	state
0	0.75	1699083168752	1699083202605	33852	0.0005755242	bl_acc	efficientnet_b0	COMPLETE
1	0.75	1699083202607	1699083235838	33230	0.0007135303	bl_acc	efficientnet_b0	COMPLETE