This train.py setup produces a segfault.