From 84290cbbf0c6e9a5de618c3fe6f452873eef24d8 Mon Sep 17 00:00:00 2001
From: KangqingYe <yekangq@sjtu.edu.cn>
Date: Sat, 19 Jun 2021 20:57:23 +0800
Subject: [PATCH] change gpu : by ykq

---
 .idea/workspace.xml | 38 +++++++++++++++++++-------------------
 main.py             | 10 +++++++++-
 solver.py           |  2 +-
 3 files changed, 29 insertions(+), 21 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 7288dad..beb6088 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -16,8 +16,8 @@
       <file pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/main.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="306">
-              <caret line="29" column="31" selection-start-line="29" selection-start-column="31" selection-end-line="29" selection-end-column="31" />
+            <state relative-caret-position="334">
+              <caret line="57" column="23" selection-start-line="57" selection-start-column="23" selection-end-line="57" selection-end-column="23" />
               <folding>
                 <element signature="e#0#15#0" expanded="true" />
               </folding>
@@ -28,8 +28,8 @@
       <file pinned="false" current-in-tab="true">
         <entry file="file://$PROJECT_DIR$/solver.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="5526">
-              <caret line="319" lean-forward="true" selection-start-line="319" selection-end-line="319" />
+            <state relative-caret-position="122">
+              <caret line="18" column="21" lean-forward="true" selection-start-line="18" selection-start-column="21" selection-end-line="18" selection-end-column="21" />
               <folding>
                 <element signature="e#0#9#0" expanded="true" />
               </folding>
@@ -40,7 +40,7 @@
       <file pinned="false" current-in-tab="false">
         <entry file="file://$PROJECT_DIR$/data_loader.py">
           <provider selected="true" editor-type-id="text-editor">
-            <state relative-caret-position="576">
+            <state relative-caret-position="-980">
               <caret line="32" selection-start-line="32" selection-end-line="32" />
               <folding>
                 <element signature="e#0#9#0" expanded="true" />
@@ -169,7 +169,7 @@
     <editor active="true" />
     <layout>
       <window_info id="Favorites" side_tool="true" />
-      <window_info active="true" content_ui="combo" id="Project" order="0" visible="true" weight="0.14464994" />
+      <window_info content_ui="combo" id="Project" order="0" visible="true" weight="0.14464994" />
       <window_info id="Structure" order="1" side_tool="true" weight="0.25" />
       <window_info anchor="bottom" id="Version Control" />
       <window_info anchor="bottom" id="Python Console" />
@@ -177,7 +177,7 @@
       <window_info anchor="bottom" id="Event Log" side_tool="true" />
       <window_info anchor="bottom" id="Message" order="0" />
       <window_info anchor="bottom" id="Find" order="1" />
-      <window_info anchor="bottom" id="Run" order="2" visible="true" weight="0.3286119" />
+      <window_info active="true" anchor="bottom" id="Run" order="2" visible="true" weight="0.3286119" />
       <window_info anchor="bottom" id="Debug" order="3" weight="0.39943343" />
       <window_info anchor="bottom" id="Cvs" order="4" weight="0.25" />
       <window_info anchor="bottom" id="Inspection" order="5" weight="0.4" />
@@ -199,16 +199,9 @@
     </breakpoint-manager>
   </component>
   <component name="editorHistoryManager">
-    <entry file="file://$PROJECT_DIR$/evaluation.py">
-      <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="36">
-          <caret line="6" column="4" selection-start-line="6" selection-start-column="4" selection-end-line="6" selection-end-column="4" />
-        </state>
-      </provider>
-    </entry>
     <entry file="file://$PROJECT_DIR$/data_loader.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="576">
+        <state relative-caret-position="-980">
           <caret line="32" selection-start-line="32" selection-end-line="32" />
           <folding>
             <element signature="e#0#9#0" expanded="true" />
@@ -216,10 +209,17 @@
         </state>
       </provider>
     </entry>
+    <entry file="file://$PROJECT_DIR$/evaluation.py">
+      <provider selected="true" editor-type-id="text-editor">
+        <state relative-caret-position="36">
+          <caret line="6" column="4" selection-start-line="6" selection-start-column="4" selection-end-line="6" selection-end-column="4" />
+        </state>
+      </provider>
+    </entry>
     <entry file="file://$PROJECT_DIR$/main.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="306">
-          <caret line="29" column="31" selection-start-line="29" selection-start-column="31" selection-end-line="29" selection-end-column="31" />
+        <state relative-caret-position="334">
+          <caret line="57" column="23" selection-start-line="57" selection-start-column="23" selection-end-line="57" selection-end-column="23" />
           <folding>
             <element signature="e#0#15#0" expanded="true" />
           </folding>
@@ -228,8 +228,8 @@
     </entry>
     <entry file="file://$PROJECT_DIR$/solver.py">
       <provider selected="true" editor-type-id="text-editor">
-        <state relative-caret-position="5526">
-          <caret line="319" lean-forward="true" selection-start-line="319" selection-end-line="319" />
+        <state relative-caret-position="122">
+          <caret line="18" column="21" lean-forward="true" selection-start-line="18" selection-start-column="21" selection-end-line="18" selection-end-column="21" />
           <folding>
             <element signature="e#0#9#0" expanded="true" />
           </folding>
diff --git a/main.py b/main.py
index 8693d48..1f0a7fb 100644
--- a/main.py
+++ b/main.py
@@ -4,6 +4,14 @@
 from data_loader import get_loader
 from torch.backends import cudnn
 import random
+import numpy as np
+
+os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+os.system('nvidia-smi -q -d Memory |grep -A4 GPU|grep Free >tmp')
+memory_gpu = [int(x.split()[2]) for x in open('tmp', 'r').readlines()]
+print('Using GPU:' + str(np.argmax(memory_gpu)))
+os.environ["CUDA_VISIBLE_DEVICES"] = str(np.argmax(memory_gpu))
+os.system('rm tmp')
 
 def main(config):
     cudnn.benchmark = True
@@ -25,7 +33,7 @@ def main(config):
     lr = 0.0002
     augmentation_prob= random.random()*0.7
     # epoch = random.choice([100,150,200,250])
-    epoch = 1
+    epoch = 1000
     decay_ratio = random.random()*0.8
     decay_epoch = int(epoch*decay_ratio)
 
diff --git a/solver.py b/solver.py
index fbe34f7..f402e4e 100644
--- a/solver.py
+++ b/solver.py
@@ -297,7 +297,7 @@ def train(self):
 				unloader = torchvision.transforms.ToPILImage()
 				SR = unloader(SR)
 				SR.save(os.path.join(self.result_path,
-														  '%s_valid_%d_SR %d.png' % (self.model_type, epoch + 1, i)))
+														  '%s_test_%d_SR %d.png' % (self.model_type, epoch + 1, i)))
 
 				#torchvision.utils.save_image(GT.data.cpu(),
 				#							 os.path.join(self.result_path,