Add swgo to main

Only messy conflict was in in this file in the train_torch function.
Not sure I resolved this correctly so we should double check here 😅.