Update with some work on Training

migueldeicaza · migueldeicaza · commit 62e2e7bd1aad · 2018-05-29T22:08:32.000-04:00
diff --git a/TensorFlowSharp/Training/Checkpointable.cs b/TensorFlowSharp/Training/Checkpointable.cs
@@ -0,0 +1,90 @@
+&#65279;//
+// Port of the checkpointable code from Python.
+//
+// Authors:
+//   Miguel de Icaza
+//
+using System;
+using System.Collections.Generic;
+
+namespace TensorFlow
+{
+	/// <summary>
+	/// Checkpointable reference.
+	/// </summary>
+	public class CheckpointableReference
+	{
+		/// <summary>
+		/// Local name for the dependency
+		/// </summary>
+		/// <value>The name.</value>
+		public string Name { get; private set; }
+
+		/// <summary>
+		/// The Checkpointable object being referenced.
+		/// </summary>
+		/// <value>The reference.</value>
+		public CheckpointableBase Reference { get; private set; }
+
+		public CheckpointableReference (string name, CheckpointableBase reference)
+		{
+			Name = name;
+			Reference = reference;
+		}
+	}
+
+	/// <summary>
+	/// Indicates a position within a Checkpoint
+	/// </summary>
+	public class CheckpointPosition
+	{
+	}
+
+	/// <summary>
+	/// Base class for `Checkpointable` objects without automatic dependencies.
+	/// </summary>
+	/// <remarks>
+	/// Dependencies must be added explicitly, unless attribute assignment 
+	/// is performance-critical use <see cref="T:TensorFlow.Checkpointable"/>
+	/// </remarks>
+	public class CheckpointableBase
+	{
+		List<CheckpointableReference> _unconditional_checkpoint_dependencies;
+		Dictionary<string, CheckpointableReference> _unconditional_dependency_names;
+		Dictionary<string, CheckpointPosition> _deferred_dependencies;
+		int update_uid;
+
+		public void MaybeInitializeCheckpointTable ()
+		{
+			// If we have already been initialized
+			if (_unconditional_checkpoint_dependencies != null)
+				return;
+			
+			_unconditional_checkpoint_dependencies = new List<CheckpointableReference> ();
+			_unconditional_dependency_names = new Dictionary<string, CheckpointableReference> ();
+			_deferred_dependencies = new Dictionary<string, CheckpointPosition> ();
+			update_uid = -1;
+		}
+
+		/// <summary>
+		/// All dependencies for this object
+		/// </summary>
+		/// <value>A list of CheckpointableReference objects indicating named Checkpointable dependencies which should be saved along with this object.</value>
+		/// <remarks>
+		/// <para>
+		/// May be overridden to include conditional dependencies.  
+		/// </para>
+		/// <para>
+		/// </para>
+		/// </remarks>
+		public virtual IList<CheckpointableReference> CheckPointDependencies => _unconditional_checkpoint_dependencies;
+
+		/// <summary>
+		/// Look up a dependency by name, may be overridden to include conditional dependencies.
+		/// </summary>
+		/// <param name="name">Name.</param>
+		public virtual CheckpointableReference LookupDependency (string name) => _unconditional_dependency_names.TryGetValue (name, out var res) ? res : null;
+
+
+	}
+}
diff --git a/TensorFlowSharp/Training/Optimizer.cs b/TensorFlowSharp/Training/Optimizer.cs
@@ -91,6 +91,9 @@ protected override object UpdateOp (Optimizer optimizer, TFGraph graph)
 		}
 	}
 
+	/// <summary>
+	/// Optimizer.
+	/// </summary>
 	public class Optimizer {
 
 		// What should the parameter be?   The Python code calls

Original file line number	Diff line number	Diff line change
`@@ -91,6 +91,9 @@ protected override object UpdateOp (Optimizer optimizer, TFGraph graph)`
`91`	`91`	`}`
`92`	`92`	`}`
`93`	`93`
	`94`	`+ /// <summary>`
	`95`	`+ /// Optimizer.`
	`96`	`+ /// </summary>`
`94`	`97`	`public class Optimizer {`
`95`	`98`
`96`	`99`	`// What should the parameter be? The Python code calls`