Unet_mha_ref_attn run training example
concrete13377 opened this issue · 0 comments
concrete13377 commented
Hello! Thanks for your work. Could you please help me understand how I can run training of unet_mha_ref_attn from this pr on viton dataset?
concrete13377 opened this issue · 0 comments
Hello! Thanks for your work. Could you please help me understand how I can run training of unet_mha_ref_attn from this pr on viton dataset?