ThunderModule
does not convert targets to the same type as outputs of the model.
#76
Labels
invalid
This doesn't seem right
questionable
It is not clear whether feature or fix is needed. Should be discussed.
In
inference_step
,y
remains the same type after being received from data loader. Howeverx
is guaranteed to benp.ndarray
after being processed by architecture.The text was updated successfully, but these errors were encountered: