|
@@ -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
|
|
opt.save_dir = increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok | opt.evolve) # increment run
|
|
|
|
|
|
# watermark save dictionary
|
|
# 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
|
|
# DDP mode
|
|
opt.total_batch_size = opt.batch_size
|
|
opt.total_batch_size = opt.batch_size
|