StudentModel.py
# Initialize model
student_model = StudentNet()

# Define optimizer
student_optimizer = optim.Adam(student_model.parameters(),
                               lr=0.001)