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