yuantianyuan01 / StreamMapNet

GNU General Public License v3.0
189 stars 16 forks source link

Model saving #5

Closed alfredgu001324 closed 1 year ago

alfredgu001324 commented 1 year ago

Hi, I saw that in your config file you use IterBased runner. I am wondering why this is preferred over epoch based? Also, I am wondering when the intermediate model will be saved? i.e. if the performance increase or?

alfredgu001324 commented 1 year ago

Is it possible to put it in a EpochBased runner format so that we can see the improvements over epochs? Thanks!

yuantianyuan01 commented 1 year ago

IterBasedRunner is more flexible and can easily support temporal training. You may try modifing the code to use EpochBasedRuuner if necessary. The checkpoint saving schedule are defined here.

alfredgu001324 commented 1 year ago

@yuantianyuan01 Thanks for your clarification! Would modifying to EpochBased runner degrade the performance?

yuantianyuan01 commented 1 year ago

It shouldn't affect the performance unless it brings additional bugs.

alfredgu001324 commented 1 year ago

I see, thank you so much!