transformers
c74befc9 - HFTracer.trace can now take callables and torch.nn.Module (#18457)

Commit
3 years ago
HFTracer.trace can now take callables and torch.nn.Module (#18457) * Enable HFTracer to trace with custom dummy inputs instead of pre-computed ones * Add HFTracer.trace docstring, and make it possible to handle callable and torch.nn.Module in general * Remove pdb comment * Apply suggestions
Parents
Loading