ndgnuh / megane

1 stars 0 forks source link

Idea(s) for multi-scheme models #3

Open ndgnuh opened 1 year ago

ndgnuh commented 1 year ago
ndgnuh commented 1 year ago
ndgnuh commented 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)