diff --git a/docs/tutorials/gluon/custom_layer.md b/docs/tutorials/gluon/custom_layer.md index 40ba0823a9d5..706b8455a4cd 100644 --- a/docs/tutorials/gluon/custom_layer.md +++ b/docs/tutorials/gluon/custom_layer.md @@ -156,7 +156,7 @@ class NormalizationHybridLayer(gluon.HybridBlock): self.scales = self.params.get('scales', shape=scales.shape, - init=mx.init.Constant(scales.asnumpy().tolist()), # Convert to regular list to make this object serializable + init=mx.init.Constant(scales.asnumpy()), differentiable=False) def hybrid_forward(self, F, x, weights, scales): diff --git a/docs/tutorials/python/profiler.md b/docs/tutorials/python/profiler.md index 91a74e4f49cf..06d3a8db7f56 100644 --- a/docs/tutorials/python/profiler.md +++ b/docs/tutorials/python/profiler.md @@ -195,6 +195,7 @@ print(profiler.dumps()) You can also dump the information collected by the profiler into a `json` file using the `profiler.dump()` function and view it in a browser. ```python +profiler.set_state('run') profiler.dump(finished=False) ```