part2
part2 copied to clipboard
Cudaing too much
Hey, what's up? Thanks so much for uploading the code.
regarding this line:
def Var(*sz): return Parameter(Arr(*sz)).cuda()
I believe cudaing this decoder parameters is redundant as we are cudaing the decoder later on in the code.