Slight updates

leschultz · Apr 19, 2024 · 49c0385 · 49c0385
1 parent 01f6413
commit 49c0385
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/examples/synthetic/fit.py b/examples/synthetic/fit.py
@@ -42,6 +42,7 @@ def main():
                       n_epochs=n_epochs,
                       batch_size=batch_size,
                       lr=lr,
+                      patience=10,
                       save_dir=save_dir,
                       )
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 # Package information
 name = 'multilearn'
-version = '0.0.0'  # Need to increment every time to push to PyPI
+version = '0.0.1'  # Need to increment every time to push to PyPI
 description = 'Multi-task learning with Pytorch.'
 url = 'https://github.com/leschultz/multilearn'
 author = 'Lane E. Schultz'

diff --git a/tests/test_fit.py b/tests/test_fit.py
@@ -13,7 +13,7 @@ def test_ml(self):
         save_dir = 'outputs'
         lr = 1e-4
         batch_size = 32
-        n_epochs = 10
+        n_epochs = 2
         tasks = ['toy1', 'toy2', 'friedman1']
 
         # Data