• Julien Plu's avatar
    TF version of the trainer (#4017) · aad50151
    Julien Plu authored
    * First commit to add a TF version of the trainer.
    
    * Make the TF trainer closer to what looks the PT trainer
    
    * Refactoring common code between the PT and TF trainer into an util file.
    
    * Some bugfix + better similarity with the PT trainer
    
    * Add missing class in transformers init
    
    * Bugfix over prediction + use classification report instead of simple metrics
    
    * Fix name error
    
    * Fix optimization tests + style
    
    * Apply style
    
    * Several bugfix for multi-gpu training
    
    * Apply style
    
    * Apply style
    
    * Add glue example for the TF trainer
    
    * Several bugix + address the reviews
    
    * Fix on the TF training args file
    
    * Add a debug mode
    
    * Bugfix in utils_ner.py when segment_ids is None
    
    * Apply style
    
    * Apply style
    
    * Add TPU strategy
    
    * Fix selection strategy
    aad50151
test_optimization_tf.py 3.49 KB