Update lpips.py

richzhang · Sep 10, 2021 · c82f1ed · c82f1ed
1 parent ecfae6f
commit c82f1ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lpips/lpips.py b/lpips/lpips.py
@@ -50,8 +50,8 @@ def __init__(self, pretrained=True, net='alex', version='0.1', lpips=True, spati
  [True] is for test mode (default)
  [False] is for training mode
  pnet_tune
- [False] tune the base/trunk network
- [True] keep base/trunk frozen
+ [False] keep base/trunk frozen
+ [True] tune the base/trunk network
  use_dropout : bool
  [True] to use dropout when training linear layers
  [False] for no dropout when training linear layers