modifing the Scripts for any arbitrary output size

IntelLabs / bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch

BSD 3-Clause "New" or "Revised" License

544 stars 73 forks source link

modifing the Scripts for any arbitrary output size #4

Closed narminGhaffari closed 3 years ago

narminGhaffari commented 3 years ago

I am trying to use these repo for classification with 3 output nodes. However, in the fully connected layer, I got the error of size mismatch. Can you help me with this issue?

ranganathkrishnan commented 3 years ago

Can you give more details, or code snippet of your model? It is not clear what is the issue/error that you are facing.

ranganathkrishnan commented 3 years ago

This seem to be related to custom model development and information on the issue is not provided to help further. Closing the issue since no response on requested details.