We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e7ef57f commit b5ac6caCopy full SHA for b5ac6ca
src/lightning/pytorch/callbacks/finetuning.py
@@ -31,8 +31,13 @@
31
import lightning.pytorch as pl
32
from lightning.pytorch.callbacks.callback import Callback
33
from lightning.pytorch.utilities.exceptions import MisconfigurationException
34
+from lightning.pytorch.utilities.imports import _TORCHVISION_AVAILABLE
35
from lightning.pytorch.utilities.rank_zero import rank_zero_warn
36
37
+if not _TORCHVISION_AVAILABLE:
38
+ __doctest_skip__ = ["BackboneFinetuning"]
39
+
40
41
log = logging.getLogger(__name__)
42
43
0 commit comments