Running drums_rnn_train hangs on calling checkpoint listeners on first checkpoint

Question

I\'m trying to train a magenta model on a set of hi hat MIDI patterns and upon running

drums_rnn_train --config=\'one_drum\' --run_dir=/tmp/drums_rnn/logdir/run2