resnet50OnFashionMnistDataset
fun resnet50OnFashionMnistDataset()
Content copied to clipboard
This example shows how to do image classification from scratch using pre-made resnet50Light model, without leveraging pre-trained weights. We demonstrate the workflow on the FashionMnist classification dataset.
It includes:
dataset loading from S3
usage of pre-defined model from org.jetbrains.kotlinx.dl.api.core.model package
model compilation
model training
model evaluation