Просмотр исходного кода

修改投影矩阵保存位置

liyan 1 год назад
Родитель
Сommit
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