mirror of
https://github.com/clearml/clearml
synced 2025-03-03 18:52:12 +00:00
Fix FastAI DeprecationWarning (#683)
This commit is contained in:
parent
85c27fda82
commit
717edba8c2
@ -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()
|
||||||
|
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
fastai >= 2.0.0
|
fastai >= 2.6.0
|
||||||
tensorboard
|
tensorboard
|
||||||
tensorboardX
|
tensorboardX
|
||||||
clearml
|
clearml
|
||||||
|
Loading…
Reference in New Issue
Block a user