2025 USA-NA-AIO Round 2, Problem 1, Part 8

AI Olympiads 2025 USA-NA-AIO Round 2
1

Part 8 (5 points, coding task)

In this part, you are asked to configure your optimizer.

  1. Define an optimizer object called optimizer.

  2. Configure the optimization method as Adam.

  3. Set the learning rate as 1e-3.

2 
### WRITE YOUR SOLUTION HERE ###

lr = 1e-3
optimizer = optim.Adam(model.parameters(), lr=lr)

""" END OF THIS PART """