Use a small network, train it in a local GPU