函数作用:更新后重置梯度
(self, curr_loss=None)
| 938 | self.flush_gradients() |
| 939 | |
| 940 | def flush_gradients(self, curr_loss=None): |
| 941 | """ |
| 942 | 函数作用:更新后重置梯度 |
| 943 | """ |
| 944 | for k, v in self.layers.items(): |
| 945 | v.flush_gradients() |
| 946 | |
| 947 | def fit(self, X_train, y_train, n_epochs=20, batch_size=64, verbose=False, epo_verbose=True): |
| 948 | """ |
no outgoing calls
no test coverage detected