Thx. I think I wrote codes w/o considering autograd in-place condition (now it looks like numpy or tf rather than torch). Hope I don’t need to redesign the code but just replacing in-place ops with torch supported equivalents for now