Skip to content

Instantly share code, notes, and snippets.

@zaleslaw
Created June 9, 2022 12:33
Show Gist options
  • Save zaleslaw/db97ffe4d997927dcd5f1ef059f14f98 to your computer and use it in GitHub Desktop.
Save zaleslaw/db97ffe4d997927dcd5f1ef059f14f98 to your computer and use it in GitHub Desktop.
Top model description
val topModel = Sequential.of(
GlobalAvgPool2D(
name = "top_avg_pool",
),
Dense(
name = "top_dense",
kernelInitializer = GlorotUniform(),
biasInitializer = GlorotUniform(),
outputSize = 200,
activation = Activations.Relu
),
Dense(
name = "pred",
kernelInitializer = GlorotUniform(),
biasInitializer = GlorotUniform(),
outputSize = NUM_CLASSES,
activation = Activations.Linear
),
noInput = true
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment