Skip to content

Instantly share code, notes, and snippets.

@zaleslaw
Created June 9, 2022 13:03
Show Gist options
  • Save zaleslaw/506a1606a774339720b4f78a8bd4e9f4 to your computer and use it in GitHub Desktop.
Save zaleslaw/506a1606a774339720b4f78a8bd4e9f4 to your computer and use it in GitHub Desktop.
How to use mutiple callbacks
model.use {
it.compile(
optimizer = Adam(clipGradient = ClipGradientByValue(0.1f)),
loss = Losses.SOFT_MAX_CROSS_ENTROPY_WITH_LOGITS,
metric = Metrics.ACCURACY
)
it.logSummary()
it.fit(
dataset = train,
epochs = EPOCHS,
batchSize = TRAINING_BATCH_SIZE,
callbacks = listOf(earlyStopping, terminateOnNaN)
)
val accuracy = it.evaluate(
dataset = test,
batchSize = TEST_BATCH_SIZE,
callback = EvaluateCallback()
).metrics[Metrics.ACCURACY]
val predictions = it.predict(
dataset = test,
batchSize = TEST_BATCH_SIZE,
callback = PredictCallback()
)
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment