Complete the training and get the result.

nerdslab-club / cl_model

All models required for curious learner

GNU Affero General Public License v3.0

1 stars 0 forks source link

Closed afmjoaa closed 6 months ago

afmjoaa commented 7 months ago

[ ] Fix the inference for GPU
- model not loading correctly, output token classification head saving mis-match
[ ] Make the data loader save the the data in a json file, which can be reloader after ward.

afmjoaa commented 6 months ago

Done