Error training TorchModel without augs

[alpha-tango]

all 3 augmentations were set to False. Setting all 3 to true resolves the error.

Traceback (most recent call last):
  File "/mydir/acodet/run.py", line 83, in <module>
    main()
  File "/mydir/acodet/run.py", line 70, in main
    run_training(**kwargs)
  File "/mydir/acodet/acodet/train.py", line 364, in run_training
    model = train(model, data_loaders, model_output_dir, device=conf.DEVICE)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mydir/acodet/acodet/torch_train.py", line 105, in train
    outputs = model(
              ^^^^^^
  File "/mydir/acodet/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1776, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mydir/acodet/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1787, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mydir/acodet/acodet/torchmodels.py", line 82, in forward
    x_spec = self.augment(x_spec, noise_spec, y, plot=plot)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mydir/acodet/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1776, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mydir/acodet/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1787, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mydir/acodet/acodet/torch_augment.py", line 116, in forward
    x = torch.vstack([x, x_sa, x_na, x_ts])
                         ^^^^
UnboundLocalError: cannot access local variable 'x_sa' where it is not associated with a value