ITADN

Show requires_grad info for torch tensors and parameters.

#59Pull Requestdanieldjohnson 创建于 2025-02-17已合并
D
danieldjohnsoncommented
For `Tensor`s, we add requires_grad=True to the summary when that flag is set on the tensor. For `nn.Parameter`s, we assume gradients are required by default and add requires_grad=False when the parameter is frozen. Summary shows regardless of whether or not autovizualization is on. Fixes #51.
合并状态:已合并 合并于 2025-02-17 关闭于 2025-02-17 0 条评论