FSDNet icon indicating copy to clipboard operation
FSDNet copied to clipboard

Run on the PyTorch 1.3+

Open xw-hu opened this issue 3 years ago • 2 comments

  1. nets.py: return irnn()(input, self.up_weight.weight....) -> return irnn().forward(input, self.up_weight.weight....)
  2. irnn.py: @cupy.util.memorize(...) -> @cupy.memorize(...)

xw-hu avatar Jul 29 '21 11:07 xw-hu