search

Huage001 / AdaAttN

Officially unofficial PyTorch re-implementation of paper: AdaAttN: Revisit Attention Mechanism in Arbitrary Neural Style Transfer, ICCV 2021.
Apache License 2.0
210 stars 34 forks source link

Q. Traning time #9

Closed sunwoo76 closed 2 years ago

sunwoo76 commented 2 years ago

The results of your paper is interesting.

So, I want to train your model from scratch.

Before training, I want to know approximate training time when you trained your model.

Thanks.

Huage001 commented 2 years ago

Thank you for your interest. It takes less than 1 day on a Tesla P40 GPU to train the model 5 epochs on COCO dataset with batch size 8.