Tweaked finetune/pretrain script to pass the SGD method param