Fix FastAI DeprecationWarning (#683)

This commit is contained in:
Rizwan Hasan 2022-05-30 22:56:37 +06:00 committed by GitHub
parent 85c27fda82
commit 717edba8c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -13,7 +13,7 @@ try:
get_image_files, get_image_files,
ImageDataLoaders, ImageDataLoaders,
Resize, Resize,
cnn_learner, vision_learner,
resnet34, resnet34,
error_rate, error_rate,
) )
@ -34,7 +34,7 @@ def main(epochs):
dls = ImageDataLoaders.from_name_func(path, files, label_func, item_tfms=Resize(224), num_workers=0) dls = ImageDataLoaders.from_name_func(path, files, label_func, item_tfms=Resize(224), num_workers=0)
dls.show_batch() dls.show_batch()
learn = cnn_learner(dls, resnet34, metrics=error_rate) learn = vision_learner(dls, resnet34, metrics=error_rate)
learn.fine_tune(epochs, cbs=[TensorBoardCallback()]) learn.fine_tune(epochs, cbs=[TensorBoardCallback()])
learn.show_results() learn.show_results()

View File

@ -1,4 +1,4 @@
fastai >= 2.0.0 fastai >= 2.6.0
tensorboard tensorboard
tensorboardX tensorboardX
clearml clearml