image_output_1.png
test = CustomImageList.from_csv_custom(path=path, csv_name='test.csv', imgIdx=0)

data = CustomImageList.from_csv_custom(path=path, csv_name='train.csv', imgIdx=1).split_by_rand_pct(0.2).label_from_df(cols='label').add_test(test, label=0).transform(get_transforms(do_flip=False)).databunch(bs=128, num_workers=0).normalize(imagenet_stats)

learn = cnn_learner(data, models.resnet18, metrics=[accuracy], model_dir='/kaggle/working/models')

learn.lr_find()
learn.recorder.plot(suggestion=True)