Detaching STN input - Githubissues

aicaffeinelife / Pytorch-STN

Spatial Transformer Networks in Pytorch.

MIT License

155 stars 35 forks source link

Detaching STN input #6

Open janzd opened 4 years ago

janzd commented 4 years ago

Hello, can I ask you a question? Why the input of conv1 in SpatialTransformer should be detached? Look forward to your reply.

Originally posted by @wuwenbin970731 in https://github.com/aicaffeinelife/Pytorch-STN/issues/5#issuecomment-659167815

janzd commented 4 years ago

That's a good question. I don't know why it's there but I implemented STN myself based on this repository without detaching the input and it seems to work fine.

wuwenbin970731 commented 4 years ago

Thanks for your reply, I will try it without detaching the input.