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,
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