List or Tuple as output?

Chris-hughes10 / Yolov7-training

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.

GNU General Public License v3.0

116 stars 35 forks source link

List or Tuple as output? #10

Open jmcginn opened 1 year ago

jmcginn commented 1 year ago

PyTorch trace likes to have Tuples as output but the model currently uses lists. Can this be changed without breaking things?

Chris-hughes10 commented 1 year ago

I would think that it can probably be changed, but it isn't something that I've tested. @bepuca any thoughts?