Ejemplos de get_dataset_classes en Python

Ejemplo n.º 1

Archivo: params.py Proyecto: zwcdp/ParlAI

 def add_pyt_dataset_args(self, opt):
     """Add arguments specific to specified pytorch dataset"""
     dataset_classes = get_dataset_classes(opt)
     for dataset, _, _ in dataset_classes:
         try:
             if hasattr(dataset, 'add_cmdline_args'):
                 dataset.add_cmdline_args(self)
         except argparse.ArgumentError:
             # already added
             pass

Ejemplo n.º 2

 def add_pyt_dataset_args(self, opt):
     """Add arguments specific to specified pytorch dataset."""
     from parlai.core.pytorch_data_teacher import get_dataset_classes
     dataset_classes = get_dataset_classes(opt)
     for dataset, _, _ in dataset_classes:
         try:
             if hasattr(dataset, 'add_cmdline_args'):
                 dataset.add_cmdline_args(self)
         except argparse.ArgumentError:
             # already added
             pass