Hi,
I am training a model for multi-class classification . But, I am getting really irrelgular output, I also have tried changing the learning rate but it the effect is still the same.
image874×602 32.4 KB
Below is my code -

  
      gist.github.com
  
  
    https://gist.github.com/droid786/4069fe9cb16e909eb09ed087b427fa26 1
code

import os
import torch
import numpy as np
from torchvision import datasets
import torchvision.transforms as transforms
import random

import torch.optim as optim

This file has been truncated. show original




  
  
    
    
  
  


I am really not able to figure out how to try solving it.