Conversation
converting the input to the model to float to avoid numpy array errors
|
I don't think this should be hardcoded. Maybe a model expects a tensor to have an integer datatype (e.g. a mask or index input). |
|
I agree it should not be directly converted to float32, however, it should not leave it as is either since pytorch does not support all numpy types, it seems all the unsigned types cause error, so we can either do float32 right now, or explicitly write a mapping for unsupported types. |
|
But |
converting the input to the model to float to avoid numpy array errors from torch
@oeway