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-28 12:58:31 +02:00 committed by GitHub
parent 88e877734c
commit 2f8880f0d2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
functions/make_cnn_v2.py
View file

@ -98,8 +98,8 @@ def make_cnn(
logger.info(f"Replace bias in conv2d {i} with {filename_load}") logger.info(f"Replace bias in conv2d {i} with {filename_load}")
temp = np.load(filename_load) temp = np.load(filename_load)
assert torch.equal( assert torch.equal(
torch.Tensor(temp.shape), torch.tensor(temp.shape, dtype=torch.int),
torch.Tensor(cnn[-1]._parameters["bias"].shape), torch.tensor(cnn[-1]._parameters["bias"].data.shape, dtype=torch.int),
) )
cnn[-1]._parameters["bias"] = torch.tensor( cnn[-1]._parameters["bias"] = torch.tensor(
temp, temp,
@ -116,8 +116,8 @@ def make_cnn(
logger.info(f"Replace weights in conv2d {i} with {filename_load}") logger.info(f"Replace weights in conv2d {i} with {filename_load}")
temp = np.load(filename_load) temp = np.load(filename_load)
assert torch.equal( assert torch.equal(
torch.Tensor(temp.shape), torch.tensor(temp.shape, dtype=torch.int),
torch.Tensor(cnn[-1]._parameters["weight"].shape), torch.tensor(cnn[-1]._parameters["weight"].data.shape, dtype=torch.int),
) )
cnn[-1]._parameters["weight"] = torch.tensor( cnn[-1]._parameters["weight"] = torch.tensor(
temp, temp,