Error on Validation

[zubairbaqai]

Hi , i am training the model on semantic segmentation but upon validation i get this error

File "/home/ubuntu/Dataset/pythonEnv/env/lib/python3.8/site-packages/mmseg/core/evaluation/metrics.py", line 124, in total_intersect_and_union
total_area_intersect += area_intersect
File "/home/ubuntu/Dataset/pythonEnv/env/lib/python3.8/site-packages/apex/amp/wrap.py", line 60, in wrapper
raise NotImplementedError('Do not know how to handle ' +
NotImplementedError: Do not know how to handle these types to promote: {'FloatTensor', 'DoubleTensor'}

I am not sure, if this is related to mmseg or apex. any ideas ?
i am using latest version of mmseg and Apex, maybe any suggestions on versions to be used, if its due to version incompatibility ?

Thanks