SAE: added pixel loss option.

2019-01-25 09:56:40 +04:00
parent 72646becd1
commit ef3cf392c5
2 changed files with 23 additions and 13 deletions
@@ -106,14 +106,8 @@ class ModelBase(object):
            self.options.pop('target_epoch') 
           
        self.batch_size = self.options['batch_size']
-
-        self.sort_by_yaw = self.options['sort_by_yaw']
-        if not self.sort_by_yaw:
-            self.options.pop('sort_by_yaw') 
-        
+        self.sort_by_yaw = self.options['sort_by_yaw']        
        self.random_flip = self.options['random_flip']
-        if self.random_flip:
-            self.options.pop('random_flip') 
        
        self.src_scale_mod = self.options['src_scale_mod']
        if self.src_scale_mod == 0: