Lingkai-Kong / SDE-Net

Code for paper: SDE-Net: Equipping Deep Neural network with Uncertainty Estimates
Apache License 2.0
107 stars 18 forks source link

ConcatConv2D explanation #5

Closed philipperemy closed 4 years ago

philipperemy commented 4 years ago

Can you explain what did you want to achieve here?

https://github.com/Lingkai-Kong/SDE-Net/blob/a69a7e35d0a1b509761795bd69f67066781d674b/MNIST/models/sdenet_mnist.py#L62

Is it just to add the time t as a feature to the CNN layer?

Thanks!

Lingkai-Kong commented 4 years ago

Yes. f(x,t) is also a function of t. Therefore, we use the time as an additional channel.

philipperemy commented 4 years ago

Ok thank you. Looks good.