Fix jmiller656#1

Author: jmiller656

model.py

@@ -71,7 +71,7 @@ def __init__(self,img_size=32,num_layers=32,feature_size=256,scale=2,output_chan
 		convolution layer for the computational efficiency.'
 
 		"""
-		scaling_factor = 1 if feature_size <=64 else 0.1
+		scaling_factor = 0.1
 
 		#Add the residual blocks to the model
 		for i in range(num_layers):
@@ -140,7 +140,7 @@ def set_data_fn(self,fn,args,test_set_fn=None,test_set_args=None):
 	"""
 	Train the neural network
 	"""
-	def train(self,iterations=1000,save_dir='saved_models'):
+	def train(self,iterations=1000,save_dir="saved_models"):
 		#Removing previous save directory if there is one
 		if os.path.exists(save_dir):
 			shutil.rmtree(save_dir)