discopat.nn_training.detr

Modules

collate

criterion

DETR criterion class.

detr_trainer

engine

Train and eval functions used in main.py.

matcher

Compute the matching cost and solve the corresponding LSAP.

nested_tensor