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

AI Olympiads 2025 USA-NA-AIO Round 2
1

Part 5 (5 points, coding task)

In this part, you are asked to define a Dataloader object called dataloader_PDE:

  1. dataset = dataset_train_PDE

  2. batch_size = 32

  3. shuffle = True

2 
### WRITE YOUR SOLUTION HERE ###

batch_size_PDE = 32

dataloader_PDE = DataLoader(dataset_train_PDE, batch_size=batch_size_PDE, shuffle=True)

""" END OF THIS PART """