HiDT icon indicating copy to clipboard operation
HiDT copied to clipboard

I try to write the training phase, but there are some problems occurring

Open Sweetclover opened this issue 4 years ago • 5 comments

Sweetclover avatar Sep 01 '20 06:09 Sweetclover

You can ask about the problems here, we will be happy to help

belkakari avatar Sep 02 '20 09:09 belkakari

After computing the loss value, the oldest styles are removed from the pool to keep its size at T.

Can I choose 8 as the value of T?

Sweetclover avatar Sep 03 '20 07:09 Sweetclover

In the paper, we've used T=100. This was not an issue in terms of memory since we've stored only style vectors of relatively small size (3 for daytime and 12 for wikiart), but you definitely are welcome to experiment with the hyperparameters

belkakari avatar Sep 03 '20 10:09 belkakari

initial learning rate 0:0001 for both generators and discriminators, halving the learning rate every 200000 iterations.

But what's the total number of your iterations during training?

Sweetclover avatar Sep 03 '20 12:09 Sweetclover

For the dataset of landscapes:

We trained the translation model for 450 thousand iterations with batch size four on a single NVIDIA Tesla P40.

denkorzh avatar Sep 03 '20 12:09 denkorzh