NaN weights after rotating and slicing

[sriyachakravarthy]

Hi!
The state dictionary has weights as shown post slicing and rotating. Are we initializing layers pre/post slicing?

[mxjmtxrm]

Hi, have you solved this problem?