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

add sampler for sliding-window tranformer

parent fbbb3826
No related branches found
No related tags found
No related merge requests found
......@@ -37,7 +37,7 @@ def create_sampler(
if architecture == "autoencoder":
return AutoencoderSampler(**kwargs)
elif architecture in ("encoder_only", "fast", "pytorch"):
elif architecture in ("encoder_only", "fast", "pytorch", "sliding-window"):
return EncoderOnlySampler(**kwargs)
elif architecture in ("fast-recurrent"):
return RecurrentSampler(**kwargs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment