edu icon indicating copy to clipboard operation
edu copied to clipboard

add "splatting" of logging of fc sizes to LitCNN (and MLP)

Open charlesfrye opened this issue 3 years ago • 1 comments

the conv channel sizes are "splatted" in the config -- logged as conv.1.size -- this would be good to add to the models.

i think a callback would be good -- adding a trawl through the modules to the "magic" WandbCallback for example

charlesfrye avatar Jul 19 '21 19:07 charlesfrye

this is done for LitMLP, but not LitCNN

charlesfrye avatar Aug 13 '21 01:08 charlesfrye