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