microsoft/MeshTransformer

About the code in modeling_metro.py

haolyuan opened this issue · 1 comments

Thanks for your releasing! When I run run_metro_bodymesh.py using 3DPW data, I meet a problem. It occurs in modeling_metro.py. I search the source and find it has wrong in row 437, layer_outputs = layer_module(hidden_states, attention_mask, head_mask[i]) inmodeling_bert.py. Specificially, it reports head_mask[i]=None. As a newer in transformer, I have no idea where the head_mask is difined in your code. Could you help me about my confusion?

Thanks for your releasing! When I run run_metro_bodymesh.py using 3DPW data, I meet a problem. It occurs in modeling_metro.py. I search the source and find it has wrong in row 437, layer_outputs = layer_module(hidden_states, attention_mask, head_mask[i]) inmodeling_bert.py. Specificially, it reports head_mask[i]=None. As a newer in transformer, I have no idea where the head_mask is difined in your code. Could you help me about my confusion?

I have solved this issue~