Hi,
This looks like a fairly standard model.
Just a few questions

What is the value of depth, num_filters?
What is the size of your inputs in width and height?
This output = self.num_filters*(2**i) looks like it’s going to grow very very fast no? Is that expected to be exponential in the depth?