image_output_1.png
path = Path('../input/heroes/heroes/')
path.ls()

data = ImageDataBunch.from_folder(path, train='.', valid_pct=0.1, ds_tfms=get_transforms(max_warp=0, flip_vert=True, do_flip=True), size=128, bs=16).normalize(imagenet_stats)

learn = create_cnn(data, models.resnet50, metrics=accuracy, model_dir='/tmp/model/')
learn.lr_find()

learn.fit_one_cycle(5)

learn.save('overwatch-stage-1')

learn.unfreeze()
learn.fit_one_cycle(2)

learn.lr_find()

learn.fit_one_cycle(10, max_lr=slice(1e-06, 0.0001))