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

Published 20 hours ago • verified publisher icon CompVis

How to load my custom dataset when training

Open WocMaker opened this issue 3 years ago • 2 comments

Hi guys, I'm trying to fine-tune a VQ_GAN from imagenet with my own dataset. However I was quite confused by this point." Creating 2 text file pointing at your folder". Does it mean that I just copy command provided in the readme? Then where shall I put these two txt file? Can anyone help me out with it? AX9PG(Z05VZRDW$GBM2WF{M

WocMaker avatar Jul 15 '21 07:07 WocMaker

The training and test file locations are defined here: https://github.com/CompVis/taming-transformers/blob/master/configs/custom_vqgan.yaml#L36 and https://github.com/CompVis/taming-transformers/blob/master/configs/custom_vqgan.yaml#L41

adeptflax avatar Jul 18 '21 17:07 adeptflax

There are the path of images in the txt file.

CreamyLong avatar Dec 28 '23 14:12 CreamyLong