pretrain

参数统计

def get_parnum(m):
    total = sum([param.nelement() for param in m.parameters()])
    print("Number of parameter: %.2fM" % (total / 1e6))

results matching ""

    No results matching ""