Remove self reference

zhangxiaopang88 · Mar 17, 2020 · 7ffd9c0 · 7ffd9c0
1 parent 6ad5095
commit 7ffd9c0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/init_Wi.py b/init_Wi.py
@@ -52,13 +52,13 @@
 
 #init W_i
 print('Initializing Discriminator weights')
-if not os.path.isdir(self.path_to_Wi):
-    os.mkdir(self.path_to_Wi)
-for i in tqdm(range(num_videos)):
-    if not os.path.isfile(self.path_to_Wi+'/W_'+str(i)+'/W_'+str(i)+'.tar'):
+if not os.path.isdir(path_to_Wi):
+    os.mkdir(path_to_Wi)
+for i in tqdm(range(num_vid)):
+    if not os.path.isfile(path_to_Wi+'/W_'+str(i)+'/W_'+str(i)+'.tar'):
         w_i = torch.rand(512, 1)
-        os.mkdir(self.path_to_Wi+'/W_'+str(i))
-        torch.save({'W_i': w_i}, self.path_to_Wi+'/W_'+str(i)+'/W_'+str(i)+'.tar')
+        os.mkdir(path_to_Wi+'/W_'+str(i))
+        torch.save({'W_i': w_i}, path_to_Wi+'/W_'+str(i)+'/W_'+str(i)+'.tar')
 
 """Training"""
 batch_start = datetime.now()