pytorch_pfn_extras.onnx.annotate

Functions

annotate(**attrs)	Annotation parameters to the target function.
apply_annotation(fn, *args, **attrs)	Annotation applier to the target function
init_annotate(model, opset_ver)
scoped_anchor(**attrs)	Add anchor node to the scoped modules

pytorch_pfn_extras.nn.LazyBatchNorm3d pytorch_pfn_extras.dataset.SharedDataset