entity-neural-network / incubator

Collection of in-progress libraries for entity neural networks.
Apache License 2.0
29 stars 10 forks source link

Checkpointing #221

Closed cswinter closed 2 years ago

cswinter commented 2 years ago

Allows specifying a --checkpoint-dir to which the training state will be persisted after every iteration. If a run is interrupted, it can be fully restored from the checkpoint directory. Also makes it possible to load policies after training has completed.