torchok icon indicating copy to clipboard operation
torchok copied to clipboard
verified publisher icon eora-ai

Representation Metrics don't support bfloat16

Open Animatory opened this issue 3 years ago ā€¢ 0 comments

During training with bfloat16 metrics that use numpy inside fail. Converting types in metrics to torch.float32 would be a solution

Animatory avatar Aug 05 '22 10:08 Animatory