nv-tlabs / ATISS

Code for "ATISS: Autoregressive Transformers for Indoor Scene Synthesis", NeurIPS 2021
Other
258 stars 55 forks source link

Size missmatch in linear layer when training on Bedrooms #37

Open MunzerDw opened 1 year ago

MunzerDw commented 1 year ago

Hello!

First of all, thank you very much for publishing your code. I have preprocessed the data and want to start training, just as shown in the README. When I start training on bedroom room types, I am getting the following error:

Screenshot 2023-07-24 193805

This is happening at the "fc_class" linear layer in the "BaseAutoregressiveTransformer" module. The other room types work fine, and I am able to train with them (there is an overfitting issue though). Do you know why the error above is happening? What can I change in the configuration file (i.e. network config) to fix it?

Thank you in advance for your help.

Best, Munzer