joonmyung 1.5.4__tar.gz → 1.5.6__tar.gz

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

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

{joonmyung-1.5.4 → joonmyung-1.5.6}/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.6}/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.6}/joonmyung/script.py

@@ -3,6 +3,7 @@ from tqdm import tqdm
 import subprocess
 import time
 import pynvml
+import requests
 import datetime
 
 class GPU_Worker():
@@ -86,6 +87,23 @@ class GPU_Worker():
         for gpu in gpus.split(','):
             self.runGPUs[int(gpu)] = p
 
+    def waitForEnd(self):
+        count = 0
+        while self.check_process():
+            if self.p:
+                count += 1
+                print(f"{count}  |  CURRENT RUNNING GPUS : {list(self.runGPUs.keys())}")
+            time.sleep(self.waitTime)
+        return
+
+    def message(self):
+        url = "https://hooks.slack.com/services/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"
+        payload = {"text": "Experiments Finished"}
+        headers = {'Content-type': 'application/json'}
+
+        response = requests.post(url, json=payload, headers=headers)
+
+        return response
 
 def Process_Worker(processes, gpuWorker, p = True):
     # TODO : 실험이 완전히 끝난 시간 체크할 필요가 존재함
@@ -100,6 +118,7 @@ def Process_Worker(processes, gpuWorker, p = True):
         p = subprocess.Popen(prefix + process + suffix, shell=True)
         gpuWorker.register_process(gpus, p)
     gpuWorker.waitForEnd()
+    gpuWorker.message()
 
     end = time.localtime()
     print("------ End Running!!   : {} ------".format(time2str(end)))

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

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

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

@@ -3,7 +3,7 @@ from setuptools import find_packages
 
 setuptools.setup(
     name="joonmyung",
-    version="1.5.4",
+    version="1.5.6",
     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