pixel2style2pixel
pixel2style2pixel copied to clipboard
eladrich
Question about training custom data
Hi, thanks for your code
I want to train PSP on my own paired dataset, and the dataset is independent of the face.
After I read README and some related questions, I made the following changes:
then, I successfully run the training code:
My model is still in training. I have some questions:
- Although the code is running, I'm not sure if my settings are correct.
- I don't understand the difference between "training the PSP encoder" and other orders, do I only need to use the "training the PSP encoder" order on my dataset.
The settings look fine. I would recommend running with more than 0 workers in order to speed up training time considerably. If you wish to train an encoder (i.e, perform inversion), then you are correct that you want to follow the "training the pSp encoder" portion of the README.
Hi, i trained an encoder on my datasets, and i think the result is abnormal.
this is the result on training datasets which has 556 pair of images.
Is my training set too small?
Hi @fxcdl , It seems like you tried uploading some images but the images were not uploaded. In general, 500 images does seem a bit low for training these types of models.
Hi @fxcdl , It seems like you tried uploading some images but the images were not uploaded. In general, 500 images does seem a bit low for training these types of models.
aha, is it uploading now?
One option that can help here is to add the moco-based loss to training (using moco_lambda) and if that doesn't help, you can try strong inversion techniques such as restyle, hyperstyle, or PTI, all of which are available and compatible with SG2.