• driazati's avatar
    Add tests for results in script vs eager mode (#1430) · 227027d5
    driazati authored
    * Add tests for results in script vs eager mode
    
    This copies some logic from `test_jit.py` to check that a TorchScript'ed
    model's outputs are the same as outputs from the model in eager mode.
    
    To support differences in TorchScript / eager mode outputs, an
    `unwrapper` function can be provided per-model.
    
    * Fix inception, use PYTORCH_TEST_WITH_SLOW
    
    * Update
    
    * Remove assertNestedTensorObjectsEqual
    
    * Add PYTORCH_TEST_WITH_SLOW to CircleCI config
    
    * Add MaskRCNN unwrapper
    
    * fix prec args
    
    * Remove CI changes
    
    * update
    
    * Update
    
    * remove expect changes
    
    * Fix tolerance bug
    
    * Fix breakages
    
    * Fix quantized resnet
    
    * Fix merge errors and simplify code
    
    * DeepLabV3 has been fixed
    
    * Temporarily disable jit compilation
    227027d5
common_utils.py 12.9 KB