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
2 changed files with 3 additions and 3 deletions

View File

@@ -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()

View File

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