About spatial transform module

[Howtyee]

While learning your code, I noticed that there doesn't seem to be the spatial transform module mentioned in the text, has this module been removed from the code?

[naderalfares]

has this issue been resolved? I noticed it missing in the PyTorch implementation