Step 1
Instrument
Add Stormlog to the workload you care about, from lightweight decorators to deeper session-based profiling.
from stormlog import profile
@profile(track_tensors=True, detect_leaks=True)
def train_epoch(model, dataloader):
for batch in dataloader:
loss = model(batch)
loss.backward()






