The input ist are RGB images:
3 x 128 128
The 2d filters are:
filters: '64, 128, 256, 384'
kernel-sizes: '3, 3, 3, 3, 3'
strides: '2, 2, 2, 2, 2'

The FC that maps from flattened convs to a compressed rep:
int: 12288, out: 6144, 

The FC stream (I am using a variant of linear layers, so I need more memory), each has dimensions:
'2x in: 6044, out: 2048; in: 2048, out: 512'

And another FC:
In: 1024, Out: 8400