June 21, 2022, 11:15 a.m. | /u/Capable-Effective-93

Neural Networks, Deep Learning and Machine Learning www.reddit.com

I have been trying to use the pre-trained model in PyTorch to do some classification but only in 10 classes. However, I don't know how to change the last layer and train the model again.

Now I am considering on creating the model from scratch. However, in the paper, it seems like they have 3 softmax to ensure that after some layers, some classification is done. These are only used for training. Can I get away with not adding the …

neuralnetworks

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

Senior Engineer - Data Science Operations

@ causaLens | London - Hybrid, England, United Kingdom

F0138 - LLM Developer (AI NLP)

@ Ubiquiti Inc. | Taipei

Staff Engineer, Database

@ Nagarro | Gurugram, India

Artificial Intelligence Assurance Analyst

@ Booz Allen Hamilton | USA, VA, McLean (8251 Greensboro Dr)