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