replaced a =+ by += in order to avoid the RuntimeError (0 frames) when some movie in the training set is not used (which should be perfectly allowed).