Closed Quoding closed 3 years ago
Fix deprecation issue in MSELoss (reduction is the proper keyword) Fix issue where squeezing x in lstm decoder would result in a 1D Tensor, thus returning an error on torch.mm Add device optional keyword to quick_train
device
Also, format the code via Black.
Fix deprecation issue in MSELoss (reduction is the proper keyword) Fix issue where squeezing x in lstm decoder would result in a 1D Tensor, thus returning an error on torch.mm Add
deviceoptional keyword to quick_trainAlso, format the code via Black.