Add full equivariance model

[dgcnz]

Relevant changes:

• [x] Adds full equivariant model with config model/wang2022/full.yaml and example config at experiments/wang2022/equivariance_test/full.yaml
• [x] Adds autoregressive_train flag to wang2022 module

  
  class Wang2022LightningModule(LightningModule):
  def __init__(
      self,
      net: torch.nn.Module,
      optimizer: torch.optim.Optimizer,
      scheduler: torch.optim.lr_scheduler,
      compile: bool,
      autoregressive_train: bool = True,
  ) -> None: