Solved by Nikronic in post #2 
              
              
                Hi, 
The error says you have float input for channel size in conv layers. Note that in python, int/int=float. So whenever you used channel / 4 the output is 16.0 not 16 and that is not a valid value for channel size in conv layers. 

To solve this issue, just use // instead of / or convert to int ex…