Merge branch 'main' into feature/reports

aimhubio · Jul 16, 2024 · def426b · def426b
2 parents eb41edb + 204d4fe
commit def426b
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 13 deletions.
diff --git a/.github/workflows/nightly-release.yml b/.github/workflows/nightly-release.yml
@@ -51,7 +51,9 @@ jobs:
 
       - name: code style checks
         run: |
-          flake8 aim
+          set -e
+          ruff check .
+          ruff format . --check
 
       - name: unit-tests
         run: |

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,6 @@
 # Changelog
 
-## Unreleased
+## 3.23.0 Jul 15, 2024
 
 ### Enhancements:
 - Relax `numpy` version upper bound to include `numpy<3` (judahrand)

diff --git a/aim/VERSION b/aim/VERSION
@@ -1 +1 @@
-3.22.0
+3.23.0
diff --git a/aim/cli/init/commands.py b/aim/cli/init/commands.py
@@ -24,7 +24,7 @@ def init(repo, yes, skip_if_exists):
         elif yes:
             re_init = True
         elif skip_if_exists:
-            click.echo('Repo exists at {}. Skipped initialization.'.format(repo.root_path))
+            click.echo('Repo exists at {}. Skipped initialization.'.format(repo_path))
             return
         else:
             re_init = click.confirm(

diff --git a/aim/web/ui/package.json b/aim/web/ui/package.json
@@ -1,6 +1,6 @@
 {
   "name": "ui_v2",
-  "version": "3.22.0",
+  "version": "3.23.0",
   "private": true,
   "dependencies": {
     "@aksel/structjs": "^1.0.0",

diff --git a/examples/pytorch_track.py b/examples/pytorch_track.py
@@ -12,8 +12,8 @@
 # Initialize a new Run
 aim_run = Run()
 
-# Device configuration
-device = torch.device('cpu')
+# moving model to gpu if available
+device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 
 # Hyper parameters
 num_epochs = 5
@@ -118,8 +118,7 @@ def forward(self, x):
 
 
 # Test the model
-model.eval()
-with torch.no_grad():
+with torch.inference_mode():
     correct = 0
     total = 0
     for images, labels in test_loader:

diff --git a/examples/pytorch_track_images.py b/examples/pytorch_track_images.py
@@ -13,8 +13,8 @@
 # Initialize a new Run
 aim_run = Run()
 
-# Device configuration
-device = torch.device('cpu')
+# moving model to gpu if available
+device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 
 # Hyper parameters
 num_epochs = 5
@@ -122,8 +122,7 @@ def forward(self, x):
 
 
 # Test the model
-model.eval()
-with torch.no_grad():
+with torch.inference_mode():
     correct = 0
     total = 0
     for images, labels in tqdm(test_loader, total=len(test_loader)):