This question has been solved from my previous question below.

  
    
    
    Am I training the model correctly? 
  
  
    Hello. I am newbie for PyTorch. 
I think I fall in love with PyTorch these days, but I don’t know why XD. 
I coded softmax classifier. But I am not sure it is trained correctly. 
I get weird accuracy and loss values from the plot. 
My model code is as follows: 
import numpy as np

import torch

use_cuda = torch.cuda.is_available()

import torch.nn as nn

import torch.optim as optim

import torch.nn.functional as F

class embedding_classifier(nn.Module):

    def __init__(self, input_shape, num_c…
  


Thank you @ ptrblck.