Fixed bug in train_superimposed.py.