example_input_array dtype
Currently assumed that example_input_array dtype to be equal to model dtype. This is not necessarily correct - e.g. if input is a vector of INT.

  
    
      pytorch-lightning/pytorch_lightning/core/memory.py
    
    
         Line 192
      in
      7dc58bd
    
  
  
    

        
          
           input_ = apply_to_collection(input_, torch.Tensor, lambda x: x.type(model.dtype))