Avon0905

Results 1 comments of Avon0905

change _out_m_batch.detach().numpy()_ to _out_m_batch.detach().cpu().numpy()_ will fix this error