Fixing some comments, renaming some variables.
Fixing some comments, renaming some variables.

yozw/torch-totem,dm-jrae/torch-totem,GeorgOstrovski/torch-totem,fzvinicius/torch-totem,mevGDM/torch-totem,deepmind/torch-totem,clementfarabet/torch-totem,fbesse/torch-totem