# Initialize model
teacher_model = TeacherNet()

# Define optimizer
teacher_optimizer = optim.Adam(teacher_model.parameters(),
                               lr=0.001)

# Define loss function
teacher_criterion = nn.CrossEntropyLoss()