deep-learning-containers
deep-learning-containers copied to clipboard
[feature-request] update tensorboard profiler test
Tensorboard profiler test here
assumes that the profile logs are saved in "/logs/train/plugins/profile"
, this however might change across different tensorflow releases. As an example, before tf2.7, the profiler logs are saved in "/logs/train/plugins/profile", but in tf2.7, the logs are written in "/logs/plugins/profile". A fix for it is provided in this commit https://github.com/aws/deep-learning-containers/pull/1422/commits/ed98e7758a38843515b5aa68d9c81ff3d733aa84
A more robust test that search profiler log using regex in the logs root directory "/logs" is probably a better alternative