advertorch.bpda

BPDA

BPDAWrapper Wrap forward module with BPDA backward path If forwardsub is not None, then ignore backward

Detailed description

class advertorch.bpda.BPDAWrapper(forward, forwardsub=None, backward=None)[source]

Wrap forward module with BPDA backward path If forwardsub is not None, then ignore backward

Parameters:
  • forwardsub – substitute forward function for BPDA
  • backward – substitute backward function for BPDA