search

rxwei / DeepLearning

DEPRECATED. See tensorflow/swift-apis
https://github.com/tensorflow/swift-apis
Apache License 2.0
13 stars 6 forks source link

High-level training loop #8

Open tanmayb123 opened 5 years ago

tanmayb123 commented 5 years ago

I'm working on a model trainer that supports per-epoch and per-batch callbacks, progress bars, mini-batching, etc. Basically, just a higher-level wrapper around the training loops in the examples in https://github.com/tensorflow/swift-models

Is this important, and is this aligned with the vision of where this high-level library wants to go?

rxwei commented 5 years ago

It'd be nice to see your prototype. Progress bars would be very useful :)

BTW, the repo has moved to https://github.com/tensorflow/swift-apis.