programme.maj

// Program 2: Load a pretrained model and evaluate it on a test dataset
pretrainedModel = loadModel("model.pth")
configure {
    Dataset testDataset = loadData("test.csv")
    testDataset.split(0.8)
    testDataset.generateDataLoader()
    pretrainedModel.setLoss(crossEntropy)
    pretrainedModel.evaluate(testDataset)
}