Function print_arch

utils/__init__.py:240–242

(model, model_name='model')

def print_arch(model, model_name='model'):
    print(f"| {model_name} Arch: ", model)
    num_params(model, model_name=model_name)


def num_params(model, print_out=True, model_name="model"):