Merge pull request #940 from XinChCh/sg

update training script for bloodmnist