joonmyung 1.5.5__tar.gz → 1.5.7__tar.gz

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

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

{joonmyung-1.5.5 → joonmyung-1.5.7}/joonmyung/data.py

@@ -1,3 +1,5 @@
+from torchvision.transforms import InterpolationMode
+from torchvision import transforms
 import torch
 
 def rangeBlock(block, vmin=0, vmax=5):
@@ -26,8 +28,6 @@ def normalization(t, type = 0):
     elif type == 1:
         return t / t.min()
 
-from torchvision import transforms
-from torchvision.transforms import InterpolationMode
 
 def getTransform(train = False, totensor = False, resize=True):
 

{joonmyung-1.5.5 → joonmyung-1.5.7}/joonmyung/draw.py

@@ -13,10 +13,12 @@ import pandas as pd
 import numpy as np
 import torch.optim
 import torch
+import copy
 import cv2
 import PIL
 import os
 
+
 def drawHeatmap(matrixes, col=1, title=[], fmt=1, p=False,
                 vmin=None, vmax=None, xticklabels=False, yticklabels=False,
                 linecolor=None, linewidths=0.1, fontsize=30, r=[1,1],
@@ -341,7 +343,6 @@ def overlay(imgs, attnsL, dataset="imagenet"):
             results.append(result)
     return results
 
-import copy
 def unNormalize(image, dataset="imagenet"):
     # images : (B, C, H, W)
     if dataset == "imagenet":

{joonmyung-1.5.5 → joonmyung-1.5.7}/joonmyung/file.py

@@ -1,4 +1,3 @@
-
 import json
 import os
 import pickle

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

@@ -1,11 +1,10 @@
-import pickle
-
 from joonmyung.utils import is_dist_avail_and_initialized
 from collections import defaultdict, deque
 from joonmyung.draw import data2PIL
 from joonmyung.utils import to_np
 import torch.distributed as dist
 import datetime
+import pickle
 import wandb
 import torch
 import time

{joonmyung-1.5.5 → joonmyung-1.5.7}/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,8 +87,25 @@ 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 Process_Worker(processes, gpuWorker, p = True):
+    def message(self, text):
+        url = "https://hooks.slack.com/services/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"
+        payload = {"text": text}
+        headers = {'Content-type': 'application/json'}
+
+        response = requests.post(url, json=payload, headers=headers)
+
+        return response
+
+def Process_Worker(processes, gpuWorker, id = "", p = True):
     # TODO : 실험이 완전히 끝난 시간 체크할 필요가 존재함
     start = time.localtime()
     print("------ Start Running!! : {} ------".format(time2str(start)))
@@ -100,11 +118,14 @@ def Process_Worker(processes, gpuWorker, p = True):
         p = subprocess.Popen(prefix + process + suffix, shell=True)
         gpuWorker.register_process(gpus, p)
     gpuWorker.waitForEnd()
-
     end = time.localtime()
     print("------ End Running!!   : {} ------".format(time2str(end)))
     training_time = datetime.timedelta(seconds=time.mktime(end) - time.mktime(start))
     print(f"Time 1/all :  {training_time}/{training_time / len(processes)} ------")
+    gpuWorker.message(f"Experiments Finished" 
+                      f"{id} : "
+                      f"Time 1/all : {training_time}/{training_time / len(processes)}"
+                      )
 
 
 

{joonmyung-1.5.5 → joonmyung-1.5.7}/joonmyung/status.py

@@ -1,4 +1,3 @@
-
 import numpy as np
 import GPUtil
 import random

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

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

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

@@ -3,7 +3,7 @@ from setuptools import find_packages
 
 setuptools.setup(
     name="joonmyung",
-    version="1.5.5",
+    version="1.5.7",
     author="JoonMyung Choi",
     author_email="pizard@korea.ac.kr",
     description="JoonMyung's Library",