Giters

IntelLabs / bayesian-torch

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

modifing the Scripts for any arbitrary output size

narminGhaffari opened this issue · comments

commented

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?

commented

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

commented

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.