taming-transformers icon indicating copy to clipboard operation
taming-transformers copied to clipboard

Training transformer on custom data configuration file is missing

Open RAHULPRRAHUL opened this issue 1 year ago • 1 comments
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,

RAHULPRRAHUL avatar Sep 12 '24 17:09 RAHULPRRAHUL