See if you can keep it running, or at least keep it running longer, by partly or completely closing the choke. That would identify it as a fuel issue. If it makes no difference, the next thing to try would be an in-line spark tester, so you can monitor the spark while it is misbehaving.
