Commit 2d37ce0b authored Mar 25, 2023 by narugo1992

dev(narugo): use dropout=0.2

parent fff9d2c2

zoo/monochrome/levit2d.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -152,7 +152,7 @@ class LeViT(nn.Module):
		stages=3,
		dim_key=32,
		dim_value=64,
		dropout=0.,
		dropout=0.2,
		num_distill_classes=None
		):
		super().__init__()