[fix] call init_attention_mask inside of training loop#76

ebsmothers merged 1 commit intometa-pytorch:mainmeta-pytorch/torchforge:mainfrom

ebsmothers:init-flex-maskebsmothers/forge:init-flex-maskCopy head branch name to clipboard