Closed lhnguyen102 closed 3 months ago
This PR fixed the bugs raised in #54
input_states.to_device()
I only tested MixtureReluCuda. However, the rest activation funs should work. Let me know if it is the case.
MixtureReluCuda
close #54
Description
This PR fixed the bugs raised in #54
Changes Made
input_states.to_device()
in the forward passNote for Reviewers
I only tested
MixtureReluCuda
. However, the rest activation funs should work. Let me know if it is the case.close #54