few-shot-gan icon indicating copy to clipboard operation
few-shot-gan copied to clipboard

How to train?

Open atugar opened this issue 5 years ago • 5 comments

Can you please explain what the below params mean? They are not needed for StyleGAN 2.

python run_training.py
--config=config-ada-sv-flat
--data-dir=/path/to/datasets
--dataset-train=path/to/train \ - What is this? --dataset-eval=path/to/eval \ - What is this? --total-kimg=30
--metrics=None

atugar avatar Oct 24 '20 19:10 atugar

Ok. I read the readme again. When you say split dataset into training and eval, are those directories supposed to hold the raw JPG or TFRecord? And how do I specify what network to start from? I assume for 1024 this will be FFHQ config-f, but where to specify this?

atugar avatar Oct 24 '20 19:10 atugar

Hi, so they should both be paths to TFRecord dirs. You first create the TFRecord dirs from images using the dataset_tool.py script, and then run training with paths to those dirs. --resume-pkl-dir controls which path pickles are found in and --resume-pkl' controls the name, but if this is not given it is inferred based on the name of --dataset-train`. I'll update the readme with these details.

e-271 avatar Oct 26 '20 15:10 e-271

Hello. Thanks for getting back to me. If --dataset-train is the path to the training dir, then what is --data-dir? I thought that was the path to the training dir.

atugar avatar Oct 26 '20 15:10 atugar

I am also unclear on what data-dir is, and would appreciate clarification!

AndreTInfante avatar Oct 28 '20 21:10 AndreTInfante

@atugar I actually figured it out. data-dir is the parent directory for the datasets, and data-train and data-eval are subdirectories relative to that parent directory

AndreTInfante avatar Oct 28 '20 22:10 AndreTInfante