Spaces:
Running
on
Zero
Running
on
Zero
File size: 512 Bytes
8483373 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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"
|