Skip to content

Commit

Permalink
Merge pull request #41 from JuliaReach/schillic/format
Browse files Browse the repository at this point in the history
Format code
  • Loading branch information
schillic authored May 4, 2024
2 parents acc4bc2 + f4299c5 commit 5f72b7e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/FileFormats/ONNX.jl
Original file line number Diff line number Diff line change
Expand Up @@ -75,8 +75,9 @@ function read_ONNX(filename::String; input_dimension=nothing)
n_layers = div(idx - 2, 2)
# 4 operations per layer +1 for the input operation
# (-1 potentially for implicit identity activation in the last layer)
@assert length(ops) == 4 * n_layers || length(ops) == 4 * n_layers + 1 "" *
"each layer should consist of 4 operations (except possibly the last one)"
@assert length(ops) == 4 * n_layers ||
length(ops) == 4 * n_layers + 1 "each layer should consist of 4 " *
"operations (except possibly the last one)"
T = DenseLayerOp{<:ActivationFunction,Matrix{Float32},Vector{Float32}}
layers = T[]
layer = 1
Expand Down

0 comments on commit 5f72b7e

Please sign in to comment.