Net(
  (head): Linear(in_features=64, out_features=3, bias=True)
)