Update tensorboard logging to plot test time details

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification

https://classyvision.ai

MIT License

1.59k stars 278 forks source link

Update tensorboard logging to plot test time details #745

Closed mannatsingh closed 3 years ago

mannatsingh commented 3 years ago

Summary: It is helpful to know the memory consumption and speed during test phases. Removed the epoch duration since it can be derived from the images / sec numbers

Differential Revision: D28308240

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D28308240