fix

2019-05-02 00:39:18 +04:00
parent 6dc1d43fe9
commit 54fa75aa13
3 changed files with 6 additions and 3 deletions
@@ -353,7 +353,7 @@ class SAEModel(ModelBase):
            training_data_dst_path = self.training_data_dst_path
            sort_by_yaw = self.sort_by_yaw
            
-            if self.pretrain:
+            if self.pretrain and self.pretraining_data_path is not None:
                training_data_src_path = self.pretraining_data_path
                training_data_dst_path = self.pretraining_data_path
                sort_by_yaw = False