{ "in_features": 640, "out_features": 896, "bias": false, "activation_function": "torch.nn.modules.linear.Identity" }