Function for custom ResNet model looks like this:
def get_resnet( inputs, init_nf=16, init_ks=5, init_drop=.2, res_drop=.5 repeats=None,