JayYip / m3tl

BERT for Multitask Learning
https://jayyip.github.io/m3tl/
Apache License 2.0
545 stars 125 forks source link

Move function "_reshape_tensor" outside #116

Closed DLPerf closed 1 year ago

DLPerf commented 1 year ago

115 Move tf.function decorated function "_reshape_tensor" out of "reshape_tensors_in_dataset" to reduce retracing.