Added torch_dmoe defaults, bug fixes for 2D inputs #1210
defaults for torch dmoe match mb dmoe
1a52a06c
defaults for torch dmoe match mb dmoe
67ca3898
defaults for torch dmoe match mb dmoe
87660846
defaults for torch dmoe match mb dmoe
74664fca
defaults for torch dmoe match mb dmoe
49ffa615
defaults for torch dmoe match mb dmoe
fa5c1e41
defaults for torch dmoe match mb dmoe
3fcaa072
top k proper
fc27c599
permute fix
676331b3
permute fix
c9f83d36
permute fix
189e0233
permute fix
b3c9a8d0
permute fix
419ea28f
narrow down world size bug
0bc078e0
narrow down world size bug
ad3a22ad
narrow down world size bug
c34be692
narrow down world size bug
2353b7e3
narrow down world size bug
81b3c312
narrow down world size bug
250047b4
narrow down world size bug
981f5946
narrow down world size bug
a53dbb7a
narrow down world size bug
3aff4093
narrow down world size bug
d6553bef
narrow down world size bug
89fd113b
narrow down world size bug
761a942c
narrow down world size bug
f388896e
narrow down world size bug
cc750827
blocking dimension bug
1886608c
done
d6ae5188
yo
98a1443d
yo
bfd6b7c8
bigning
approved these changes
on 2024-05-15
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub