torchok icon indicating copy to clipboard operation
torchok copied to clipboard

Add requirements for each component type in src/models

Open Alicegaz opened this issue 3 years ago • 1 comments

It is not obvious how to discriminate between nn blocks in different subdirectories of src/models. And it is better to add readme and base classes for each component (block, brick, module, etc) listing all the requirements (what it accepts, what it returns, what it is allowed and supposed to do with the inputs).

Alicegaz avatar Jul 07 '22 06:07 Alicegaz

Indeed, we need a documentation for this

VladVin avatar Jul 28 '22 19:07 VladVin