changed trainVCTSM to 500 iter of SGD