请问我用的pytorch1.7.1和1.8，都一直报 TracerWarning: Encountering a list at the output of the tracer might cause the trace to be incorrect, this is only valid if the container structure does not change based on the module's inputs. Consider using a constant container instead (e.g. for `list`, use a `tuple` instead. for `dict`, use a `NamedTuple` instead). If you absolutely need this and know the side effects, pass strict=False to trace() to allow this behavior.

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。

GNU General Public License v3.0

861 stars 150 forks source link

请问我用的pytorch1.7.1和1.8，都一直报 TracerWarning: Encountering a list at the output of the tracer might cause the trace to be incorrect, this is only valid if the container structure does not change based on the module's inputs. Consider using a constant container instead (e.g. for `list`, use a `tuple` instead. for `dict`, use a `NamedTuple` instead). If you absolutely need this and know the side effects, pass strict=False to trace() to allow this behavior. #43

Open westream opened 1 year ago

bubbliiiing commented 1 year ago

无碍

f1owkang commented 1 year ago

解决办法：在callbacks.py中31行 self.writer.add_graph(model, dummy_input,use_strict_trace=False) 参考https://pytorch.org/docs/stable/generated/torch.jit.trace.html