This commit is contained in:
iperov 2019-05-02 00:39:18 +04:00
parent 6dc1d43fe9
commit 54fa75aa13
3 changed files with 6 additions and 3 deletions

View file

@ -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