Traceback (most recent call last):
File "clip.py", line 29, in <module>
traced_text_model = torch.jit.trace(text_model, text)
File "/usr/local/lib/python3.8/dist-packages/torch/jit/_trace.py", line 759, in trace
return trace_module(
File "/usr/local/lib/python3.8/dist-packages/torch/jit/_trace.py", line 976, in trace_module
module._c._create_method_from_trace(
RuntimeError: Only tensors, lists, tuples of tensors, or dictionary of tensors can be output from traced functions
我通过以上代码将pt模型转化为独立的bert和visual torchscript模型,visual模型可以正常转化,但是bert模型转换报错,请问应该怎么转换?转换错误如下: