tfl.test_utils.TimeTracker

Tracks time.

Keeps track of time spent in its scope and appends it to 'list_to_append' on exit from scope divided by 'num_steps' if provided.

training_step_times = [] with TimeTracker(training_step_times, num_steps=num_epochs): model.fit(... epochs=num_epochs ...) print np.median(training_step_times)

Methods

__enter__

View source

__exit__

View source