/fireo

Model shape debugger for torch. Think torch.summary but better

Primary LanguagePython

Watchers