Pete Tae-hoon Kim

BCHW format

13

Excellent work! I would like to use this in the middle of my pytorch network, so my tensors are in [Batch x Channel x Height x Width] format. I tried...

Hello, If I would like to add AvgPool2d to emitter, which function (like SpatialDilatedMaxPooling_updateOutput for MaxPool2d) should I call? I guess I can just copy codes from MaxPool2d and change...

AttributeError: creator

7

hello, When I try sample code (in README) to export PyTorch model, like ``` out = net.forward(torch.autograd.Variable(torch.FloatTensor(1,3,227,227))) thexport.save('pymodel.net', out) ``` I got the following error. ``` File "...... /python3.6/site-packages/torch/autograd/variable.py", line...

plan to support Sigmoid?

1

Hello! Very nice work! Do you have a plan to support Sigmoid activation? or if I would like to add it by myself, can you roughly let me know steps...