ML Training Storage Calculator