bert-experiments

bert-experiments copied to clipboard

Out-of-memory

2

Hi, I'm using a 40GB VRAM memory GPU but get out-of-memory error. The error seems to come from `src/model_bert.py`, Line 339 (alpha_f): ``` _, attention_probs, value_layer = self_outputs output_head_weights =...

thangld201