During my diploma thesis, I added a couple of extra blocks in order to demonstrate my model's architecture (in python code), so I added a simple BatchNorm and FullyConnected block with some different colors. Plus I added an example of architecture in the folder Deep4-net as a showcase.
P.S This fork was created a couple of years ago, so I am not familiar if there are any updates with the repo.
During my diploma thesis, I added a couple of extra blocks in order to demonstrate my model's architecture (in python code), so I added a simple BatchNorm and FullyConnected block with some different colors. Plus I added an example of architecture in the folder Deep4-net as a showcase.
P.S This fork was created a couple of years ago, so I am not familiar if there are any updates with the repo.