tensorflow-trainer

# tensorflow-trainer ## Overview TensorFlow/Keras model training skill with callbacks, distributed strategies, TensorBoard integration, and production-ready model export capabilities. ## Capabilities - Keras model training with callbacks - Custom training loops with tf.GradientTape - Distribution strategy configuration (MirroredStrategy, MultiWorkerMirroredStrategy, TPUStrategy) - TensorBoard logging and visualization - SavedModel export for TF Serving - TFLite conversion for edge deployment - Mixed precision training ## Target Processes - Model Training Pipeline with Experiment Tracking - Distributed Training Orchestration - Model Deployment Pipeline ## Tools and Libraries - TensorFlow - Keras - TensorBoard - TensorFlow Serving - TensorFlow Lite ## Input Schema ```json { "type": "object", "required": ["modelConfig", "dataConfig", "trainingConfig"], "properties": { "modelConfig": { "type": "object", "properties": { "modelPath": { "type": "string" }, "modelType": { "type": "string", "enum": ["sequential", "functional", "subclassed"] } } }, "dataConfig": { "type": "object", "properties": { "trainPath": { "type": "string" }, "valPath": { "type": "string" }, "batchSize": { "type": "integer" }, "prefetch": { "type": "boolean" } } }, "trainingConfig": { "type": "object", "properties": { "epochs": { "type": "integer" }, "optimizer": { "t...