You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When inference with the highway early-exit given a batch B;
when |B| = 1, the code is ok to run;
when |B| > 1, the code can corrupt in the code of modeling_highway_bert.py due to if highway_entropy < self.early_exit_entropy[i]: conditional statement ;
Only one sample can be inferred.
Therefore, the inference is not GPU friendly.
The text was updated successfully, but these errors were encountered:
When inference with the highway early-exit given a batch B;
when |B| = 1, the code is ok to run;
when |B| > 1, the code can corrupt in the code of modeling_highway_bert.py due to
if highway_entropy < self.early_exit_entropy[i]:
conditional statement ;Only one sample can be inferred.
Therefore, the inference is not GPU friendly.
The text was updated successfully, but these errors were encountered: