prateekjoshi565/define_loss_bert.py

## define_loss_bert.py
# converting list of class weights to a tensor
weights= torch.tensor(class_weights,dtype=torch.float)

# push to GPU
weights = weights.to(device)

# define the loss function
cross_entropy  = nn.NLLLoss(weight=weights)

# number of training epochs
epochs = 10
	# converting list of class weights to a tensor
	weights= torch.tensor(class_weights,dtype=torch.float)

	# push to GPU
	weights = weights.to(device)

	# define the loss function
	cross_entropy = nn.NLLLoss(weight=weights)

	# number of training epochs
	epochs = 10