Skip to content

Commit e7eaa89

Browse files
committed
Add basic optimizer support
This adds the optimizer trait and related types, as well as GradientDescentOptimizer and AdadeltaOptimizer implementations.
1 parent f7ee300 commit e7eaa89

File tree

3 files changed

+409
-1
lines changed

3 files changed

+409
-1
lines changed

src/lib.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,9 @@ mod variable;
187187
#[cfg(feature = "experimental_training")]
188188
pub use crate::variable::*;
189189

190+
#[cfg(feature = "experimental_training")]
191+
pub mod train;
192+
190193
////////////////////////
191194

192195
c_enum!("Error values that can be returned.", TF_Code, Code {

0 commit comments

Comments
 (0)