Monkey-patch torch.nn.Parameter
xl0 opened this issue · comments
Alexey Zaytsev commented
At the moment, the repr of a Parameter looks like this:
c1.bias
Parameter containing:
Parameter[64] x∈[-2.807, 1.645] μ=-0.387 σ=0.783
I feel we can get rid of the first line, because the type is clearly visible anyway.
Research what's the best way to do it. Probably monkey-patching the nn.Parameter class.