search

mead-ml / mead-baseline

Deep-Learning Model Exploration and Development for NLP
Apache License 2.0
243 stars 73 forks source link

Fix/onnx recent pytorch #929

Closed dpressel closed 2 years ago

dpressel commented 2 years ago

Recent versions of PyTorch (after 1.7.1) do not support the boolean comparison on export. It works if we change the definition of MASK_FALSE from a boolean to an int (0)