invalid syntax fixes

[ArashHosseini]

hey @mattans: -kwargs syntax fix, the code is also working absolutely fine with py-3.5 and up to torch-1.0.0(latest stable). 32 and 64 didn't work as batch size for me, the reason is well know so 128 should be fine as default , ~4gb on gpu. thx