update edutraintpl and version

HFUT-LEC · Aug 30, 2023 · b274cb9 · b274cb9
1 parent 0fee349
commit b274cb9
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 5 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -9,7 +9,7 @@
 project = 'EduStudio'
 copyright = '2023, HFUT-LEC'
 author = 'HFUT-LEC'
-release = 'v1.0.0-beta2'
+release = 'v1.0.0-beta2.1'
 
 import sphinx_rtd_theme
 import os

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -1,5 +1,5 @@
 .. EduStudio documentation master file.
-.. title:: EduStudio v1.0.0-beta2
+.. title:: EduStudio v1.0.0-beta2.1
 .. image:: assets/logo.png
 
 =========================================================

diff --git a/edustudio/__init__.py b/edustudio/__init__.py
@@ -2,4 +2,4 @@
 from __future__ import print_function
 from __future__ import division
 
-__version__ = 'v1.0.0-beta2'
+__version__ = 'v1.0.0-beta2.1'
diff --git a/edustudio/traintpl/edu_traintpl.py b/edustudio/traintpl/edu_traintpl.py
@@ -92,7 +92,9 @@ def fit(self, train_loader, valid_loader):
                 self.optimizer.zero_grad()
                 loss.backward()
                 self.optimizer.step()
-                for k in loss_dict: logs[k][batch_id] = loss_dict[k].item() if loss_dict[k] is not None else np.nan
+                for k in loss_dict: 
+                    if loss_dict[k] is not None:
+                        logs[k][batch_id] = loss_dict[k].item()
 
             for name in logs: logs[name] = float(np.nanmean(logs[name]))
 

diff --git a/setup.py b/setup.py
@@ -42,7 +42,7 @@
 
 setup(
     name="edustudio",
-    version="v1.0.0-beta2", 
+    version="v1.0.0-beta2.1", 
     description="a Unified and Templatized Framework for Student Assessment Models",
     long_description=long_description,
     python_requires='>=3.8',