Fixes for Python 3 compatibility

[codeman38]

• Use six.iteritems to ensure lazy iteration on both Python 2 and Python 3
• Use integer division when dividing epoch_size by n_save