Skip to content

Training transformer on custom data configuration file is missing #257

@RAHULPRRAHUL

Description

@RAHULPRRAHUL

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,

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions