Shang Liu
Results
11
comments of
Shang Liu
> > i have the same question...have you figure it out? 😄 > > i tried torchinfo.summary with gpt-2 and it works. but note that, when you use `torchinfo.summary(model, input_size=(...))`...