Skip to content

Commit cd74c32

Browse files
committed
add new file to output folder with information about generating process
1 parent 02ad696 commit cd74c32

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/generate_class_specific_samples.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,15 @@ def generate(self, iterations=150, blur_freq=6, blur_rad=0.8, wd = 0.05):
6666
#save final image
6767
im_path = f'../generated/class_{self.target_class}/c_{self.target_class}_iter_{i}_loss_{class_loss.data.numpy()}.jpg'
6868
save_image(self.created_image, im_path)
69+
70+
with open(f'../generated/class_{self.target_class}/run_details.txt', 'w') as f:
71+
f.write(f'Blur freq: {blur_freq}\n')
72+
f.write(f'Blur radius: {blur_rad}\n')
73+
f.write(f'Weight decay: {wd}\n')
74+
f.write(f'Iterations: {iterations}\n')
75+
f.write(f'Clip value: {clipping_value}\n')
76+
77+
os.rename(f'../generated/class_{self.target_class}', f'../generated/class_{self.target_class}_blurfreq_{blur_freq}_blurrad_{blur_rad}_wd{wd}')
6978
return self.processed_image
7079

7180

0 commit comments

Comments
 (0)