Spaces:
Running
on
Zero
Running
on
Zero
| python inversion/invert.py \ | |
| --device="cuda:0" \ | |
| --mean_path="files/mean.pt" \ | |
| --std_path="files/std.pt" \ | |
| --v_path="files/V.pt" \ | |
| --std_path="files/std.pt" \ | |
| --dim_path="files/weight_dimensions.pt" \ | |
| --imfolder="inversion/images/real_image/real/" \ | |
| --mask_path="inversion/images/real_image/mask.png" \ | |
| --epochs=400 \ | |
| --lr=1e-1 \ | |
| --weight_decay=1e-10 \ | |
| --dim=10000 \ | |
| --save_name="files/real_inversion1.pt" | |