davrot/kk_contour_net_shallow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-07-20 01:51:27 +02:00 committed by GitHub
parent 9bb3d5befc
commit 031e35b0b8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
inspect_weights_conv_0.py
View file

@ -23,7 +23,9 @@ assert layer_id < len(model)
weights = model[layer_id]._parameters["weight"].data
bias = model[layer_id]._parameters["bias"].data
weight_grid = tv.utils.make_grid(weights, nrow=8, padding=2, scale_each=scale_each)
weight_grid = tv.utils.make_grid(
weights, nrow=8, padding=2, scale_each=scale_each, pad_value=float("NaN")
)
v_max_abs = torch.abs(weight_grid[0, ...]).max()