DSE-MSU / DeepRobust

A pytorch adversarial library for attack and defense methods on images and graphs
MIT License
994 stars 192 forks source link

Fix the save path problem using `os.path.join` #105

Closed enderdzz closed 2 years ago

enderdzz commented 2 years ago

In deeprobust/image/defense/pgdtraining.py troch.save(), there is a problem with the save path splicing, because the default value of save_dir does not have '/' at the end. So I use os.path.join to fix the path splicing process. Also, I have applied this approach to other torch.save().