transformers
FX tracing improvement
#14321
Merged

Commits
  • changed the way tracing happens, enabling dynamic axes out of the box
    michaelbenayoun committed 3 years ago
  • Updated the tests and modeling xlnet
    michaelbenayoun committed 3 years ago
  • Added the non recoding of leaf modules to avoid recording more values for the methods to record than what will be seen at tracing time (which would otherwise desynchronize the recorded values and the values that need to be given to the proxies during tracing, causing errors).
    michaelbenayoun committed 3 years ago
  • Comments and making tracing work for gpt-j and xlnet
    michaelbenayoun committed 3 years ago
  • Refactored things related to num_choices (and batch_size, sequence_length)
    michaelbenayoun committed 3 years ago
  • style fix
    michaelbenayoun committed 3 years ago
  • Updated fx to work on PyTorch 1.10
    michaelbenayoun committed 3 years ago
  • Postponed autowrap_function feature usage for later
    michaelbenayoun committed 3 years ago
  • style fix
    michaelbenayoun committed 3 years ago
  • fixed issue
    michaelbenayoun committed 3 years ago
  • implemented suggestions
    michaelbenayoun committed 3 years ago
  • Add copyrights
    michaelbenayoun committed 3 years ago
  • Remove unnecessary file
    michaelbenayoun committed 3 years ago
  • Merge branch 'master' into fx_tracing_enhancement
    michaelbenayoun committed 3 years ago
  • Fix copies
    michaelbenayoun committed 3 years ago
  • Fix issue with add_new_model_like
    michaelbenayoun committed 3 years ago
  • Fix issue with add_new_model_like
    michaelbenayoun committed 3 years ago
  • Apply suggestions
    michaelbenayoun committed 3 years ago
  • Apply suggestions
    michaelbenayoun committed 3 years ago
Loading