search

victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.7k stars 1.08k forks source link

add iteration-based training #53

Closed SunQpark closed 5 years ago

SunQpark commented 5 years ago

Implementation of iteration-based training, requested in issue #50. trainer instance now take an argument len_epoch, which determines the number of steps(batches) in each epoch.

Following is utility function to repeat pytorch data_loader to loop it endlessly.

def inf_loop(data_loader):
    '''
    wrapper function to make pytorch data loader loops endlessly.
    '''
    for loader in repeat(data_loader):
        for data, target in loader:
            yield data, target
SunQpark commented 5 years ago

Added two commits which are

  1. Slightly changed inf_loop.
  2. Readme file updated