yuval-alaluf / SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754
https://yuval-alaluf.github.io/SAM/
MIT License
632 stars 151 forks source link

Inference results as a test results issue #25

Closed MetiCodePrivateLimited closed 3 years ago

MetiCodePrivateLimited commented 3 years ago

I have run successfully the command for training model. The issue I am getting is that the result for the inference on the trained images is not coming while training the model. Mean, I am getting training outputs but not test outputs.

I used the following paths: 'ffhq': '/images1024x1024', 'celeba_test': '/to'

I placed an image of an user in the 'celeba_test' folder to test while training the model but no response on the inference side.

and following command: python3 scripts/train.py \ --dataset_type=ffhq_aging \ --exp_dir=/to/experiment \ --workers=6 \ --batch_size=6 \ --test_batch_size=6 \ --workers=6 \ --test_workers=6 \ --val_interval=2000 \ --save_interval=10000 \ --start_from_encoded_w_plus \ --id_lambda=0.1 \ --lpips_lambda=0.1 \ --lpips_lambda_aging=0.1 \ --lpips_lambda_crop=0.6 \ --l2_lambda=0.25 \ --l2_lambda_aging=0.25 \ --l2_lambda_crop=1 \ --w_norm_lambda=0.005 \ --aging_lambda=5 \ --cycle_lambda=1 \ --input_nc=4 \ --max_steps=30000 \ --output_size=1024 \ --target_age=uniform_random \ --use_weighted_id_loss \ --checkpoint_path=trained/sam_ffhq_aging.pt

yuval-alaluf commented 3 years ago

All logs that are outputted during training will be saved to the directory specified by --exp_dir, which in your case is /to/experiment. In this directory you will see a logs directory with outputs during training.