--- a +++ b/loss/utils.py @@ -0,0 +1,17 @@ +import torch + + +def to_float_and_cuda(input): + # input = input.type(torch.FloatTensor) + # input = input.type(torch.LongTensor) + input = input.float() + input = input.cuda() + return input + + +def to_long_and_cuda(input): + # input = input.type(torch.FloatTensor) + # input = input.type(torch.LongTensor) + input = input.long() + input = input.cuda() + return input \ No newline at end of file