google-research / vision_transformer

Apache License 2.0
10.18k stars 1.27k forks source link

Fixes img/sec/core. #280

Open copybara-service[bot] opened 1 year ago

copybara-service[bot] commented 1 year ago

Fixes img/sec/core.

Before this fix, only lt0 but not lstep was updated after computing an evaluation, which led to a img/sec/core computation that was too high. Tis fix simply adds the time needed to eval / checkpoint to lt0, correcting the dt term in the dsteps/dt computation.