Commit 2eef1b8
committed
Finish EarlyStopping
1 parent 4672971 commit 2eef1b8
3 files changed
Lines changed: 4 additions & 2 deletions
File tree
- src
- TensorFlowNET.Core
- Keras/Layers
- Operations/NnOps
- TensorFlowNET.Keras/Callbacks
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -17,7 +17,7 @@ public interface ILayer: IWithTrackable, IKerasConfigable | |||
| 17 | 17 | List<IVariableV1> TrainableVariables { get; } | |
| 18 | 18 | List<IVariableV1> TrainableWeights { get; } | |
| 19 | 19 | List<IVariableV1> NonTrainableWeights { get; } | |
| 20 | - List<IVariableV1> Weights { get; set} | ||
| 20 | + List<IVariableV1> Weights { get; set; } | ||
| 21 | 21 | Shape OutputShape { get; } | |
| 22 | 22 | Shape BatchInputShape { get; } | |
| 23 | 23 | TensorShapeConfig BuildInputShape { get; } | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -84,6 +84,8 @@ public abstract class RnnCell : ILayer, RNNArgs.IRnnArgCell | |||
| 84 | 84 | protected bool built = false; | |
| 85 | 85 | public bool Built => built; | |
| 86 | 86 | ||
| 87 | + List<IVariableV1> ILayer.Weights { get => throw new NotImplementedException(); set => throw new NotImplementedException(); } | ||
| 88 | + | ||
| 87 | 89 | public RnnCell(bool trainable = true, | |
| 88 | 90 | string name = null, | |
| 89 | 91 | TF_DataType dtype = TF_DataType.DtInvalid, | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -102,8 +102,8 @@ public void on_epoch_end(int epoch, Dictionary<string, float> epoch_logs) | |||
| 102 | 102 | { | |
| 103 | 103 | Console.WriteLine($"Restoring model weights from the end of the best epoch: {_best_epoch + 1}"); | |
| 104 | 104 | } | |
| 105 | + _parameters.Model.Weights = _best_weights; | ||
| 105 | 106 | } | |
| 106 | - _parameters.Model.Weights = _best_weights; | ||
| 107 | 107 | } | |
| 108 | 108 | } | |
| 109 | 109 | public void on_train_end() | |
0 commit comments