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-27 20:13:59 +02:00 committed by GitHub
parent 842225ae29
commit 28a21b32c2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cnn_training.py
View file

@ -165,6 +165,8 @@ def run_network(
logger=logger,
data_path=data_path,
)
assert data_train.__len__() > 0
input_shape = data_train.__getitem__(0)[1].shape
logger.info("Loading test data")
data_test = alicorn_data_loader(
@ -252,6 +254,7 @@ def run_network(
conv_0_power_softmax=conv_0_power_softmax,
conv_0_meanmode_softmax=conv_0_meanmode_softmax,
conv_0_no_input_mode_softmax=conv_0_no_input_mode_softmax,
input_shape=input_shape,
).to(device)
logger.info(model)