• Yih-Dar's avatar
    Make TF pt-tf equivalence test more aggressive (#15839) · 923c35b5
    Yih-Dar authored
    
    
    * Make TF pt-tf equivalence test more aggressive
    
    * Fix for TFConvNextModelTest and TFTransfoXLModelTest
    
    * fix kwargs for outputs
    
    * clean-up
    
    * Add docstring for check_outputs()
    
    * remove: need to rename encoder-decoder
    
    * clean-up
    
    * send PyTorch things to the correct device
    
    * Add back the accidentally removed test case in test_pt_tf_model_equivalence()
    
    * Fix: change to tuple before calling check_outputs()
    
    * Fix: tfo could be a list
    
    * use to_tuple()
    
    * allow tfo only to be tuple or tensor
    
    * allow tfo to be list or tuple for now + style change
    
    * minor fix
    
    * remove np.copy and update comments
    
    * tfo -> tf_output, same for pt
    
    * Add more detailed comment
    
    * remove the incorrect comment
    Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
    923c35b5
test_modeling_tf_common.py 73.1 KB