소스 검색

修改投影矩阵保存位置

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