This commit is contained in:
Phil Wang
2023-03-04 20:30:02 -08:00
committed by GitHub
parent 5699ed7d13
commit f621c2b041

View File

@@ -303,7 +303,7 @@ cct = CCT(
pooling_padding = 1,
num_layers = 14,
num_heads = 6,
mlp_radio = 3.,
mlp_ratio = 3.,
num_classes = 1000,
positional_embedding = 'learnable', # ['sine', 'learnable', 'none']
)
@@ -1043,7 +1043,7 @@ cct = CCT(
pooling_padding = 1,
num_layers = 14,
num_heads = 6,
mlp_radio = 3.,
mlp_ratio = 3.,
num_classes = 1000,
positional_embedding = 'learnable'
)