Open ndgnuh opened 1 year ago
for step, (images, samples) in enumerate(loader):
optimizer.zero_grad()
outputs: FixedOutputType = model(images, post_process=True)
for i, sample in enumerate(samples)
sample.boxes = outputs.boxes[i]
sample.classes = outputs.classes[i]
sample.scores = outputs.scores[i]
fabric.backward(outputs.loss)
fabric.clip_gradients(model, optimizer, max_norm=5)
optimizer.step()
where FixedOutputType
is a convention by another modules (not trainer or model)
Sample
)