Misunderstanding size of final layer in tabular_leaner


(Dien Hoa TRUONG) #1

I’m playing around with the fast.ai library V1 and I don’t understand why the last layer doesn’t include the out_sz but the last emb_szs1. I’m running the tabular example in /example/tabular.

I think it should be 2 but not 100 right ? It is a binary classification model

When I see in 009_rossmann notebook, a regression model, the output size is 1.

Thank you in advance.


#2

It’s a bug that has been fixed. You’ll see it in the next release of the fastai library on conda/pip (or maybe already if you update).


(Dien Hoa TRUONG) #3

Thank you @sgugger. I’m currently at fast.ai 1.0.5 which is the lastest released version. Do you mean if I reinstall fast.ai version for developper the bug will be fixed ?


(Dien Hoa TRUONG) #4

[Update] I uninstalled fast.ai conda version and reinstalled the developer version following the instructions on github. It works now :smiley: Thank you so much.