Added some buffer to SoftMaxTree (for compat with cuda)
fixed SoftMaxTree.__init bug