Slim down checkpoints - Githubissues

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.

https://tryolabs.com

BSD 3-Clause "New" or "Revised" License

2.4k stars 399 forks source link

Closed joaqo closed 6 years ago

joaqo commented 6 years ago

Improve checkpoint creation, resulting in:

Much simpler init operations (much faster training start up time).
Much smaller checkpoints (eg.: FasterRCNN: 537MB -> 190MB, SSD: 1,731MB -> 103MB).

Also smaller changes: Improve SSD initialization (faster training), Improve testing on train.py