joonmyung 1.5.4__tar.gz → 1.5.5__tar.gz

{joonmyung-1.5.4 → joonmyung-1.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: joonmyung
-Version: 1.5.4
+Version: 1.5.5
 Summary: JoonMyung's Library
 Home-page: https://github.com/pizard/JoonMyung.git
 Author: JoonMyung Choi

{joonmyung-1.5.4 → joonmyung-1.5.5}/joonmyung/analysis/dataset.py

@@ -81,6 +81,13 @@ class JDataset():
             lns.append(ln)
         return torch.cat(ds, dim=0), torch.stack(ls, dim=0), lns
 
+    def getIndex(self, c: list = [0, 1000], i: list = [0, 50]):
+        [c_s, c_e], [i_s, i_e] = c, i
+        c = torch.arange(c_s, c_e).reshape(-1, 1).repeat(1, i_e - i_s).reshape(-1)
+        i = torch.arange(i_s, i_e).reshape(1, -1).repeat(c_e - c_s, 1).reshape(-1)
+        c_i = torch.stack([c, i], dim=-1)
+        return c_i
+
     def __len__(self):
         return
 

{joonmyung-1.5.4 → joonmyung-1.5.5}/joonmyung/log.py

@@ -120,10 +120,9 @@ class Logger():
         wandb.finish()
 
     def save(self, model, args, name):
-        if self.main_process:
+        if self.main_process and self.use_wandb:
             path = os.path.join(wandb.run.dir, f"{name}.pth")
-            torch.save({"model" : model, "args"  : args},
-                       path)
+            torch.save({"model" : model, "args"  : args}, path)
             wandb.save(path, wandb.run.dir)
 
 

{joonmyung-1.5.4 → joonmyung-1.5.5}/joonmyung.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: joonmyung
-Version: 1.5.4
+Version: 1.5.5
 Summary: JoonMyung's Library
 Home-page: https://github.com/pizard/JoonMyung.git
 Author: JoonMyung Choi

{joonmyung-1.5.4 → joonmyung-1.5.5}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_packages
 
 setuptools.setup(
     name="joonmyung",
-    version="1.5.4",
+    version="1.5.5",
     author="JoonMyung Choi",
     author_email="pizard@korea.ac.kr",
     description="JoonMyung's Library",
@@ -23,7 +23,8 @@ setuptools.setup(
 # git add .
 # git commit
 # git push
+
 # rm -rf ./*.egg-info ./dist/*
+# python3 setup.py sdist; python -m twine upload dist/*
 
-# python setup.py sdist; python -m twine upload dist/*
 # ID:JoonmyungChoi