hi I got the error:  Type ‘Tuple[Tensor, Dict[str, List[int]]]’ cannot be traced. Only Tensors and (possibly nested) Lists, Dicts, and Tuples of Tensors can be traced
The code is below:
writer.add_graph(flow_net, (torch.DoubleTensor(dataset.datas_vali[:2,:]), idx))