search

wandb / edu

Educational materials on deep learning by Weights & Biases
http://wandb.ai
GNU General Public License v2.0
526 stars 244 forks source link

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

Open charlesfrye opened 3 years ago

charlesfrye commented 3 years ago

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 commented 3 years ago

this is done for LitMLP, but not LitCNN