Skip to content
Snippets Groups Projects
Commit d3dbe1b4 authored by Gregor Kobsik's avatar Gregor Kobsik
Browse files

fix spelling error

parent 5a1bc41f
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
sample/token_generator/token_generator_factory.py
+ 1
− 1
View file @ d3dbe1b4
...@@ -17,7 +17,7 @@ def _create_token_generator(head, model, spatial_dim): ...@@ -17,7 +17,7 @@ def _create_token_generator(head, model, spatial_dim):
""" """
if head in ('generative_basic', 'basic', 'linear'): if head in ('generative_basic', 'basic', 'linear'):
return BasicGenerator(model.compute_logits) return BasicGenerator(model.compute_logits)
if head in ('half_conv', 'halv_conv_A'): if head in ('half_conv', 'half_conv_A'):
return BasicGenerator(model.compute_logits, 2**(spatial_dim - 1)) return BasicGenerator(model.compute_logits, 2**(spatial_dim - 1))
if head in ('single_conv', 'single_conv_A'): if head in ('single_conv', 'single_conv_A'):
return BasicGenerator(model.compute_logits, 2**spatial_dim) return BasicGenerator(model.compute_logits, 2**spatial_dim)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment