[cf6a9e]: / loss / utils.py

Download this file

17 lines (13 with data), 384 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
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