oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

Published 20 hours ago • verified publisher icon Oneflow-Inc

Can oneflow eager support amp eager mode ?

Open tiancaidagongrenwangdoudou opened this issue 1 year ago • 2 comments

Hi, I want to train a neural network with oneflow eager amp mode, but I found oneflow can't support it. When will eager amp mode be supported? I don't want to use nn.Graph mode now.

tiancaidagongrenwangdoudou avatar Aug 18 '22 06:08 tiancaidagongrenwangdoudou

Good question, maybe we will support amp eager mode in v0.9.0? @chengtbf

BBuf avatar Aug 18 '22 06:08 BBuf

Good question, maybe we will support amp eager mode in v0.9.0? @chengtbf

Thanks!

tiancaidagongrenwangdoudou avatar Aug 18 '22 06:08 tiancaidagongrenwangdoudou

已经支持amp。

BBuf avatar Sep 27 '22 01:09 BBuf