Skip to content

Commit 45f91ea

Browse files
committed
fixed backward compatibility problem with invertible networks
1 parent 4c87255 commit 45f91ea

5 files changed

+5
-5
lines changed

examples/GeneralizedMinkowski/ConstrainedNeuralNetworkSegmentation_cameraman.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@
362362
"(nx, ny) = size(image)[1:2]\n",
363363
"α = 0.3^2\n",
364364
"batchsize = 1\n",
365-
"HN = H = NetworkHyperbolic(nx, ny,n_chan_in, batchsize, architecture; α)\n",
365+
"HN = H = NetworkHyperbolic(n_chan_in, architecture; α)\n",
366366
"\n",
367367
"#Test reversibility: neural network forward prop -> inverse prop\n",
368368
"X_prev, X_curr, lgdet = HN.forward(deepcopy(image),deepcopy(image))\n",

examples/GeneralizedMinkowski/ConstrainedNeuralNetworkSegmentation_cameraman.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ n_chan_in = size(image,3)
286286
(nx, ny) = size(image)[1:2]
287287
α = 0.3^2
288288
batchsize = 1
289-
HN = H = NetworkHyperbolic(nx, ny,n_chan_in, batchsize, architecture; α)
289+
HN = H = NetworkHyperbolic(n_chan_in, architecture; α)
290290

291291
#squared distance function to a constraint set
292292
function Dist2ToSet(input,P,active_channels,example_ind)

examples/GeneralizedMinkowski/ConstrainedNeuralNetworkSegmentation_motorbike.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,7 @@ n_chan_in = size(image,3)
299299
(nx, ny) = size(image)[1:2]
300300
α = 0.3^2
301301
batchsize = 1
302-
HN = H = NetworkHyperbolic(nx, ny,n_chan_in, batchsize, architecture; α)
302+
HN = H = NetworkHyperbolic(n_chan_in, architecture; α)
303303

304304
#squared distance function to a constraint set
305305
function Dist2ToSet(input,P,active_channels,example_ind)

examples/GeneralizedMinkowski/ConstrainedNeuralNetworkSegmentation_zebra.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@
329329
"(nx, ny) = size(image)[1:2]\n",
330330
"α = 0.3^2\n",
331331
"batchsize = 1\n",
332-
"HN = H = NetworkHyperbolic(nx, ny,n_chan_in, batchsize, architecture; α)\n",
332+
"HN = H = NetworkHyperbolic(n_chan_in, architecture; α)\n",
333333
"\n",
334334
"#Test reversibility: neural network forward prop -> inverse prop\n",
335335
"X_prev, X_curr, lgdet = HN.forward(deepcopy(image),deepcopy(image))\n",

examples/GeneralizedMinkowski/ConstrainedNeuralNetworkSegmentation_zebra.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ n_chan_in = size(image,3)
269269
(nx, ny) = size(image)[1:2]
270270
α = 0.3^2
271271
batchsize = 1
272-
HN = H = NetworkHyperbolic(nx, ny,n_chan_in, batchsize, architecture; α)
272+
HN = H = NetworkHyperbolic(n_chan_in, architecture; α)
273273

274274
#squared distance function to a constraint set
275275
function Dist2ToSet(input,P,active_channels,example_ind)

0 commit comments

Comments
 (0)