Skip to content

Commit f846e0e

Browse files
committed
print self.mixstyle.__class__.__name__
1 parent 3593d8a commit f846e0e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dassl/modeling/backbone/resnet.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,9 @@ def __init__(
139139
self.mixstyle = ms_class(p=ms_p, alpha=ms_a)
140140
for layer_name in ms_layers:
141141
assert layer_name in ["layer1", "layer2", "layer3"]
142-
print(f"Insert MixStyle after {ms_layers}")
142+
print(
143+
f"Insert {self.mixstyle.__class__.__name__} after {ms_layers}"
144+
)
143145
self.ms_layers = ms_layers
144146

145147
self._init_params()

0 commit comments

Comments
 (0)