Hi, thanks for the code. Can you please provide total training time and memory consumption required to run the code? Which cards you used?