Skip to content

Commit

Permalink
Change varialbe name in Model struct: input_file => fileName
Browse files Browse the repository at this point in the history
  • Loading branch information
mgyoo86 committed Oct 30, 2024
1 parent 7f46868 commit e287b14
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/TroyonBetaNN.jl
Original file line number Diff line number Diff line change
Expand Up @@ -17,15 +17,15 @@ export Troyon_Data, Load_predefined_Troyon_NN_Models
W::Matrix{Float64} # map from input to hidden layer
V::Vector{Float64} # map from hidden layer to output
βₙ_limit::Float64=NaN # scalar output
input_file::String=""
fileName::String=""
end

@kwdef mutable struct CNN_Model
n::Int=1 # toroidal mode number
model::ORT.InferenceSession
input::Dict=Dict()
βₙ_limit::Float64=NaN # scalar output
input_file::String=""
fileName::String=""
end

@kwdef mutable struct Sample_Points
Expand Down Expand Up @@ -61,7 +61,7 @@ function Load_predefined_Troyon_NN_Models(MLP_file::String =joinpath(@__DIR__, "
end

# Read CNN file
CNN = CNN_Model(model= ORT.load_inference(CNN_file));
CNN = CNN_Model(model= ORT.load_inference(CNN_file), fileName=CNN_file);

return Troyon_Data(Sample_Points(), MLPs, CNN)
end
Expand Down

0 comments on commit e287b14

Please sign in to comment.