Finish e2e GPU memory fix #9019
Labels
area:rasa-oss 🎡
Anything related to the open source Rasa framework
area:rasa-oss/ml 👁
All issues related to machine learning
research:end-to-end
type:enhancement ✨
Additions of new features or changes to existing ones, should be doable in a single PR
The prototypical GPU memory fix has been tested and solves the problem that the GPU runs out of memory for large action sets in e2e scenarios.
Implementation issue for #8734
To make it production ready, the following things need to be addressed / Definition of done:
label_batch_size
should be turned into a boolean variablebalanced_sampling_negative_labels
to remove unnecessary complexity in the interplay betweenlabel_batch_size
andnumber_of_negative_examples
The text was updated successfully, but these errors were encountered: