(v, f)
| 43 | return v, f |
| 44 | |
| 45 | def norm(v, f): |
| 46 | v = (v - v.min())/(v.max() - v.min()) - 0.5 |
| 47 | |
| 48 | return v, f |
| 49 | |
| 50 | def getGradNorm(net): |
| 51 | pNorm = torch.sqrt(sum(torch.sum(p ** 2) for p in net.parameters())) |
no outgoing calls
no test coverage detected