Downloads · 30 days
0
stephen-newberry/pcb-via-4layer-cnn-rev1
pcb-via-4layer-cnn-rev1 is a machine learning model from stephen-newberry. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as gpl-3.0.
Model requires reshaping data in the dataset to be (100,200,60) size. Use this file as follows in Julia:
Downloads · 30 days
0
Access
Public
Updated Sep 14, 2024
Repo size
137 MB
Likes
0
Public
Click a slice to open those files.
.jld2137 MB · 100%
From the Hugging Face model README
Model requires reshaping data in the dataset to be (100,200,60) size. Use this file as follows in Julia:
model = create_parametric_model()
model_state = JLD2.load("../final_model2/trained_model.jld2", "model_state")
Flux.loadmodel!(model, model_state)
naughts=zeros(Int8, (100,200,60,1))
s0 = model(naughts)
il_of_nothing = 10 .* log.(sqrt.(s0[3,:,1].^2 + s0[4,:,1].^2))
plot(il_of_nothing, linewidth=2)
More info to follow.