parser로 폴더 만들어주기

1.

parser.add_argument("--sample_path", type=str, help="path to the sample dataset")

parser.add_argument("--ckpt_path", type=str, help="path to the ckpt dataset")

 

이후

 

args = parser.parse_args()

 

 

2.

if not os.path.exists(f"./sample_folder/{args.sample_path}"):

os.mkdir(f"./sample_folder/{args.sample_path}")

 

if not os.path.exists(f"./ckpt_folder/{args.ckpt_path}"):

os.mkdir(f"./ckpt_folder/{args.ckpt_path}")

 

3.

if i % 500 == 0:

with torch.no_grad():

print(" i = ", i)

g_ema.eval()

sample, _ = g_ema([sample_z])

utils.save_image(

sample,

#f"sample/{str(i).zfill(6)}.png",

f"sample_folder/{args.sample_path}/{str(i).zfill(6)}.png",

nrow=int(args.n_sample ** 0.5),

normalize=True,

range=(-1, 1),

)

if i % 5000 == 0:

torch.save(

{

"g": g_module.state_dict(),

"d": d_module.state_dict(),

"g_ema": g_ema.state_dict(),

"g_optim": g_optim.state_dict(),

"d_optim": d_optim.state_dict(),

"args": args,

"ada_aug_p": ada_aug_p,

},

f"ckpt_folder/{args.ckpt_path}/{str(i).zfill(6)}.pt",

)