godisboy / DRPAN

Discriminative Region Proposal Adversarial Network (DRPAN)
52 stars 9 forks source link

Ability to save & restore training checkpoints #2

Closed richardburleigh closed 5 years ago

richardburleigh commented 6 years ago

DRPAN produces amazing results, but the ability to save and restore training checkpoints would be really helpful for training on a single GPU, especially on large datasets.

jiangyufeng77 commented 5 years ago

We will add this part of the code as soon as possible.