dotnet / dotnet/machinelearning

Limited functionality for fine-tuning TF models using the TensorFlowTransform

Open
#2,348 1 comment 0 reactions 0 assignees View on GitHub
area-TensorFlow Priority:2
Dominant language
C#
Stars
9.4k
Forks
2k
Avg merge
2d 20h
Merged PRs (30d)
11

Description

The TensorFlowTransform in ML.NET provides some basic capabilities for re-training pre-trained TF models.

The table below lists down the capabilities currently supported for fine-tuning TF models. It also lists functionality that's currently not supported.

| Fine-Tuning abilities | Supported by ML.NET TensorFlowTransform |
| ---------- | -------- |
| Continue retraining on TF SavedModel, given dataset with same shape as original model | Yes |
| Continue retraining on TF SavedModel, given dataset with _different_ shape as original model | No |
| Freeze weights of all layers, except the last layer. Replace last layer of TF SavedModel with new layer based on categories of new dataset. Re-train the weights for last layer only. | No |
| Freeze weights upto any arbitrary Nth layer (given layer name). Replace final layer of TF SavedModel with new layer based on categories of new dataset. Re-train the weights from (N+1) th layer to the last layer. | No |

@zeahmed @JRAlexander

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.