search

senmaoy / RAT-GAN

A conditional GAN for text-to-image
43 stars 9 forks source link

Custom dataset #27

Closed ghost closed 4 months ago

ghost commented 11 months ago

In the prepare dataset section, the line Save captions.pickle under data_dir, where is data_dir , does anyone know?

senmaoy commented 10 months ago

It's where the datas are stored.

ghost commented 10 months ago

How to train text encoder for custom dataset ?

senmaoy commented 10 months ago

It's here: https://github.com/senmaoy/RAT-GAN/blob/main/code/pretrain_DAMSM.py

ghost commented 10 months ago

I mean the command line to train the text encoder, do you know ?

senmaoy commented 10 months ago

python pretrain_DAMSM.py But you need certain modifications