Hi Team,
After training vqgan model on custom data as per below steps:
---Put your image files in a folder custom_data folder
---Create 2 text files a xx_train.txt and xx_test.txt that point to the files in your training and test set respectively
---Adapt configs/custom_vqgan.yaml to point to these 2 files
---Run python main.py --base configs/custom_vqgan.yaml -t True --gpus 0,1 to train on two GPUs.
We need to train transformer on custom data similar to to FFHQ+ CelebA-HQ data and To do so custom_transformer.yaml configuration file is not available to use.
python main.py --base configs/custom_transformer.yaml -t True --gpus 0,
Hi Team,
After training vqgan model on custom data as per below steps:
---Put your image files in a folder custom_data folder
---Create 2 text files a xx_train.txt and xx_test.txt that point to the files in your training and test set respectively
---Adapt configs/custom_vqgan.yaml to point to these 2 files
---Run python main.py --base configs/custom_vqgan.yaml -t True --gpus 0,1 to train on two GPUs.
We need to train transformer on custom data similar to to FFHQ+ CelebA-HQ data and To do so custom_transformer.yaml configuration file is not available to use.
python main.py --base configs/custom_transformer.yaml -t True --gpus 0,