DeepSpeed
DeepSpeed copied to clipboard
use `Flops Profiler` to test `model.generate()`
Fix issue #2514
Nowadays, the Flops Profiler can test forward for transformers such as bert, but for t5, it actually uses model.generate() during inference. So how to use Flops Profiler to test model.generate() ?
@microsoft-github-policy-service agree
@CaffreyR, thanks for the PR. Can you please remove the binary files? Thanks!
Hi @tjruwase , do you mean the .DS_Store? I have deleted it.
@cli99 can you review this PR?