Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # This file contains all the main external libs we'll use
- from fastai.imports import *
- from fastai.transforms import *
- from fastai.conv_learner import *
- from fastai.model import *
- from fastai.dataset import *
- from fastai.sgdr import *
- from fastai.plots import *
- PATH = "data/dataset/"
- size=224
- arch=resnet34
- #Creating classifier
- data = ImageClassifierData.from_paths(PATH, tfms=tfms_from_model(arch, sz), bs=8)
- learn = ConvLearner.pretrained(arch, data, precompute=True)
- lrf=learn.lr_find()
- learn.sched.plot()
Add Comment
Please, Sign In to add comment