%reload_ext autoreload
%autoreload 2

import os
os.chdir('..')

#export
from nb_005 import *

data = data_from_imagefolder(Path('data/dogscats'), 
    ds_tfms=get_transforms(), tfms=imagenet_norm, size=224)

learn = ConvLearner(data, tvm.resnet34, metrics=accuracy)

learn.fit_one_cycle(6)

learn.save('rn34')

learn.load('rn34')

lr2=3e-4
learn.unfreeze()
learn.fit_one_cycle(6, slice(lr2/25,lr2), pct_start=0.05)

learn.save('rn34.2')

accuracy(*learn.TTA())

learn = ConvLearner(data, tvm.resnet50, metrics=accuracy)

learn.fit_one_cycle(6)

lr2=3e-4
learn.unfreeze()
learn.fit_one_cycle(6, slice(lr2/25,lr2), pct_start=0.05)

learn.save('rn50')

accuracy(*learn.TTA())

data = data_from_imagefolder(Path('data/dogscats'),
    ds_tfms=get_transforms(), tfms=imagenet_norm, size=224)
learn = ConvLearner(data, tvm.resnet34, metrics=accuracy)
learn.fit_one_cycle(6)
learn.unfreeze()
learn.fit_one_cycle(4, slice(1e-5,3e-4))

accuracy(*learn.TTA())