Преглед на файлове

修改投影矩阵保存位置

liyan преди 1 година
родител
ревизия
4fe5f2d3c7
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      train_embed.py

+ 1 - 1
train_embed.py

@@ -536,7 +536,7 @@ if __name__ == '__main__':
         opt.save_dir = increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok | opt.evolve)  # increment run
 
     # watermark save dictionary
-    opt.key_path = increment_path(Path(opt.project) / opt.name, exist_ok=True)
+    opt.key_path = increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok | opt.evolve)
 
     # DDP mode
     opt.total_batch_size = opt.batch_size