Documentation update

[ArlindKadra]

Auto-PyTorch/autoPyTorch/pipeline/components/training/trainer/base_trainer.py

Line 229 in 0cc6b46

swa_model: Optional[torch.nn.Module] = None,

The class documentation should be updated. As I checked, the module also is in need of a documentation update.