• anj-s's avatar
    Add benchmarks for custom training loops + tf.distribute (#6980) · 65636099
    anj-s authored
    * first version of ctl
    
    * fix indent
    
    * remove monkey patching for core
    
    * add dtype arg
    
    * fix dtype arg
    
    * add logging lib
    
    * remove compat.v1.logging
    
    * add datetime import
    
    * fix FLAGS import
    
    * add constant vals
    
    * move to using as tf import
    
    * move to using as tf import
    
    * remove steps per epoch = 1
    
    * test train and test for one step
    
    * test train and test for one step
    
    * test train and test for one step
    
    * test train and test for the entire dataset
    
    * use an iterator for test
    
    * pass tensors instead of an iterator
    
    * add stats dict
    
    * fix list declaration
    
    * fix list declaration
    
    * fix elapsed time calc
    
    * print lr at epoch boundary alone
    
    * Use regular tf import instead of compat
    
    * remove tensorboard chkpts
    
    * add correct logging import
    
    * add correct logging import
    
    * add benchmark configs
    
    * add tests and configs
    
    * add tests and configs
    
    * add keras flags import
    
    * add keras flags import
    
    * fix eval ds creation cond
    
    * return numpy value of train_loss
    
    * return numpy value of loss and acc values
    
    * add option for full eager mode
    
    * fix lint errors
    
    * add ctl flags
    
    * add ctl import
    
    * add the xla flag
    
    * enable v2 behavior in unit tests
    
    * rename dataset var
    
    * add synthetic dataset without monkey patching
    
    * add ctl local constants
    
    * add ctl local constants
    
    * change to using v2 imports
    
    * change to using v2 imports
    
    * change to using v2 imports
    
    * change to using keras synthetic input fn
    
    * remove enable_eager flag from benchmarks
    
    * remove enable_eager flag from benchmarks
    
    * remove enable_eager flag from benchmarks
    
    * add option for no distrat
    
    * add lambda for flags
    
    * remove no_func benchmarks due to OOM error
    
    * remove README
    
    * remove unused comments
    
    * remove unchanged file
    
    * remove unchanged file
    
    * remove unused drop_remainder_arg
    
    * use keras.common lr function
    
    * address PR comments
    
    * remove reference to deleted file
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * .
    
    * fix lint errors
    
    * .
    65636099
ctl_common.py 1.45 KB