• Matt's avatar
    Rewrite TensorFlow train_step and test_step (#17057) · 349f1c85
    Matt authored
    * Initial commit
    
    * Better label renaming
    
    * Remove breakpoint before pushing (this is your job)
    
    * Test a lot more in the Keras fit() test
    
    * make fixup
    
    * Clarify the case where we flatten y dicts into tensors
    
    * Clarify the case where we flatten y dicts into tensors
    
    * Extract label name remapping to a method
    349f1c85
test_modeling_tf_common.py 84.3 KB