Skip to content

Commit 5f71d75

Browse files
committed
minor refactoring
1 parent 66caef3 commit 5f71d75

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/models/detection/backbone_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -140,13 +140,13 @@ def _resnet_fpn_extractor(
140140

141141

142142
def _validate_trainable_layers(
143-
pretrained: bool,
143+
is_trained: bool,
144144
trainable_backbone_layers: Optional[int],
145145
max_value: int,
146146
default_value: int,
147147
) -> int:
148148
# don't freeze any layers if pretrained model or backbone is not used
149-
if not pretrained:
149+
if not is_trained:
150150
if trainable_backbone_layers is not None:
151151
warnings.warn(
152152
"Changing trainable_backbone_layers has not effect if "

0 commit comments

Comments
 (0)