• alope107's avatar
    Transformer partial fix (#5092) · 6f5967a0
    alope107 authored
    * Fix Transformer TPU crash in Python 2.X.
    
    - Tensorflow raises an error when tf_inspect.getfullargspec is called on
    a functools.partial in Python 2.X. This issue would be hit during the
    eval stage of the Transformer TPU model. This change replaces the call
    to functools.partial with a lambda to work around the issue.
    
    * Remove unused import from transformer_main.
    
    * Fix lint error.
    6f5967a0
transformer_main.py 25.2 KB