gitextract_av2x8cy4/ ├── .gitignore ├── README.md ├── configs/ │ ├── attributes_5.json │ ├── editing/ │ │ ├── editing_with_dialog.yml │ │ └── editing_wo_dialog.yml │ └── train/ │ ├── field_1024_bangs.yml │ ├── field_1024_beard.yml │ ├── field_1024_eyeglasses.yml │ ├── field_1024_smiling.yml │ ├── field_1024_young.yml │ ├── field_128_bangs.yml │ ├── field_128_beard.yml │ ├── field_128_eyeglasses.yml │ ├── field_128_smiling.yml │ └── field_128_young.yml ├── data/ │ ├── __init__.py │ └── latent_code_dataset.py ├── editing_quantitative.py ├── editing_with_dialog.py ├── editing_wo_dialog.py ├── environment.yml ├── language/ │ ├── accuracy.py │ ├── build_vocab.py │ ├── dataset.py │ ├── generate_feedback.py │ ├── generate_training_request.py │ ├── language_utils.py │ ├── lstm.py │ ├── preprocess_request.py │ ├── run_encoder.py │ ├── templates/ │ │ ├── attr_wise_caption_templates.json │ │ ├── feedback.json │ │ ├── gender.json │ │ ├── metadata_fsm.json │ │ ├── overall_caption_templates.json │ │ ├── pool.json │ │ ├── system_mode.json │ │ ├── user_fsm.json │ │ ├── user_old_templates.json │ │ └── vocab.json │ ├── train_encoder.py │ └── utils/ │ ├── __init__.py │ ├── eval.py │ ├── logger.py │ ├── lr_schedule.py │ ├── misc.py │ ├── numerical.py │ ├── progress/ │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── MANIFEST.in │ │ ├── README.rst │ │ ├── progress/ │ │ │ ├── __init__.py │ │ │ ├── bar.py │ │ │ ├── counter.py │ │ │ ├── helpers.py │ │ │ └── spinner.py │ │ ├── setup.py │ │ └── test_progress.py │ ├── setup_logger.py │ └── visualize.py ├── models/ │ ├── __init__.py │ ├── archs/ │ │ ├── __init__.py │ │ ├── attribute_predictor_arch.py │ │ ├── field_function_arch.py │ │ └── stylegan2/ │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── LICENSE-FID │ │ ├── LICENSE-LPIPS │ │ ├── LICENSE-NVIDIA │ │ ├── __init__.py │ │ ├── apply_factor.py │ │ ├── calc_inception.py │ │ ├── checkpoint/ │ │ │ └── .gitignore │ │ ├── convert_weight.py │ │ ├── dataset.py │ │ ├── distributed.py │ │ ├── fid.py │ │ ├── generate.py │ │ ├── inception.py │ │ ├── inversion.py │ │ ├── lpips/ │ │ │ ├── __init__.py │ │ │ ├── base_model.py │ │ │ ├── dist_model.py │ │ │ ├── networks_basic.py │ │ │ ├── pretrained_networks.py │ │ │ └── weights/ │ │ │ ├── v0.0/ │ │ │ │ ├── alex.pth │ │ │ │ ├── squeeze.pth │ │ │ │ └── vgg.pth │ │ │ └── v0.1/ │ │ │ ├── alex.pth │ │ │ ├── squeeze.pth │ │ │ └── vgg.pth │ │ ├── model.py │ │ ├── non_leaking.py │ │ ├── op/ │ │ │ ├── __init__.py │ │ │ ├── fused_act.py │ │ │ ├── fused_bias_act.cpp │ │ │ ├── fused_bias_act_kernel.cu │ │ │ ├── upfirdn2d.cpp │ │ │ ├── upfirdn2d.py │ │ │ └── upfirdn2d_kernel.cu │ │ ├── ppl.py │ │ ├── sample/ │ │ │ └── .gitignore │ │ └── train.py │ ├── base_model.py │ ├── field_function_model.py │ ├── losses/ │ │ ├── __init__.py │ │ ├── arcface_loss.py │ │ └── discriminator_loss.py │ └── utils.py ├── quantitative_results.py ├── train.py └── utils/ ├── __init__.py ├── crop_img.py ├── dialog_edit_utils.py ├── editing_utils.py ├── inversion_utils.py ├── logger.py ├── numerical_metrics.py ├── options.py └── util.py