taming-transformers
taming-transformers copied to clipboard
Training transformer on custom data configuration file is missing
trafficstars
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,