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
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