search

AliaksandrSiarohin / first-order-model

This repository contains the source code for the paper First Order Motion Model for Image Animation
https://aliaksandrsiarohin.github.io/first-order-model-website/
MIT License
14.57k stars 3.22k forks source link

save checkpoint #469

Closed Fishyy599 closed 3 years ago

Fishyy599 commented 3 years ago

Must all epoch training be completed before the checkpoints file is generated? The default epoch is 100. I still don’t see the checkpoints file after more than ten generations of training.

AliaksandrSiarohin commented 3 years ago

This is controlled by checkpoint_freq param, which is 50 by default. So you should see first checkpoint after 50 epochs.

Fishyy599 commented 3 years ago

Thanks! I just read that checkpoint_freq defaults to 100. I can change the value of checkpoint_freq so that I can easily see the checkpoint.