Pārlūkot izejas kodu

修改投影矩阵保存位置

liyan 1 gadu atpakaļ
vecāks
revīzija
4fe5f2d3c7
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  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