weights2weights / inversion /run_inversion.sh
multimodalart's picture
Upload 200 files
8483373 verified
raw
history blame
512 Bytes
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"