No, it didn’t went to eval(), its just done with the train(), before going to eval(), error thrown.
I just chaged the def train() params, set def validate() as it is. am I incorrect?
imagenet script organization:
def main()
def main_worker()
def train()
   outputs, aux_outputs = model(inputs)
   loss1 = criterion(outputs, target)
   loss2 = criterion(aux_outputs, target)
   loss = loss1 + 0.4*loss2
def validate()
   outputs = model(inputs)
   loss = criterion(outputs, target)
def adjust_learning_rate()
def accuracy()