From 717edba8c2b39fb7486bd2aba9ca0294f309b4c3 Mon Sep 17 00:00:00 2001 From: Rizwan Hasan Date: Mon, 30 May 2022 22:56:37 +0600 Subject: [PATCH] Fix FastAI DeprecationWarning (#683) --- examples/frameworks/fastai/fastai_with_tensorboard_example.py | 4 ++-- examples/frameworks/fastai/requirements.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/frameworks/fastai/fastai_with_tensorboard_example.py b/examples/frameworks/fastai/fastai_with_tensorboard_example.py index 888a8352..35ded082 100644 --- a/examples/frameworks/fastai/fastai_with_tensorboard_example.py +++ b/examples/frameworks/fastai/fastai_with_tensorboard_example.py @@ -13,7 +13,7 @@ try: get_image_files, ImageDataLoaders, Resize, - cnn_learner, + vision_learner, resnet34, 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.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.show_results() diff --git a/examples/frameworks/fastai/requirements.txt b/examples/frameworks/fastai/requirements.txt index 8ed8a05c..ee930fd2 100644 --- a/examples/frameworks/fastai/requirements.txt +++ b/examples/frameworks/fastai/requirements.txt @@ -1,4 +1,4 @@ -fastai >= 2.0.0 +fastai >= 2.6.0 tensorboard tensorboardX clearml