"comfy/ldm/vscode:/vscode.git/clone" did not exist on "982338b9bb41301000ddac46d67103af9d0582cd"
  • Julien Rossi's avatar
    Trainer with Iterable Dataset (#7858) · a09fe140
    Julien Rossi authored
    * fix 5990
    
    * accomodate iterable dataset without predefined length
    * set it as 1 use case: provide max_steps, and NO num_epochs
    * Is a merge of master and PR 5995
    
    * fix trainer test under TF
    
    * fix only for torch
    * TF trainer untouched
    * trainer tests are skipped when no torch
    
    * address comments
    
    * fix quality checks
    
    * remove torch.dataset from test_trainer
    
    * unnecessary inheritance
    * RegressionDataset implements all needed methods __len__ and __getitem__
    
    * fix quality checks
    
    * restore RegressionDataset
    
    * was wrongly under is_torch_available()
    a09fe140
test_trainer.py 27.3 KB