Ejemplo n.º 1
0
 def get_optional_params():
   return dict(DataLayer.get_optional_params(), **{
       'num_parallel_calls': int,
       'shuffle_buffer': int,
       'image_size': int,
       'num_classes': int,
   })
Ejemplo n.º 2
0
 def get_optional_params():
     return dict(
         DataLayer.get_optional_params(), **{
             'augmentation': dict,
             'pad_to': int,
             'max_duration': float,
             'bpe': bool,
             'autoregressive': bool,
         })
Ejemplo n.º 3
0
 def get_optional_params():
     return dict(
         DataLayer.get_optional_params(), **{
             'repeat': int,
             'num_cpu_cores': int,
             'tgt_vocab_file': str,
             'pad_data_to_eight': bool,
             'batch_in_tokens': bool,
         })
Ejemplo n.º 4
0
 def get_optional_params():
   return dict(DataLayer.get_optional_params(), **{
       'augmentation': dict,
       'pad_to': int,
       'max_duration': float,
       'bpe': bool,
       'autoregressive': bool,
       'syn_enable': bool,
       'syn_subdirs': list,
       'window_size': float,
       'window_stride': float,
   })
Ejemplo n.º 5
0
 def get_optional_params():
     return dict(
         DataLayer.get_optional_params(), **{
             'use_targets': bool,
             'delimiter': str,
             'target_file': str,
             'map_parallel_calls': int,
             'prefetch_buffer_size': int,
             'pad_lengths_to_eight': bool,
             'pad_vocab_to_eight': bool,
             'shuffle_buffer_size': int,
             'special_tokens_already_in_vocab': bool,
             'use_start_token': bool,
         })
Ejemplo n.º 6
0
 def get_optional_params():
   return dict(DataLayer.get_optional_params(), **{
     'data_root': str,
     'rand_start': bool,
     'small': bool,
     'use_targets': bool,
     'delimiter': str,
     'map_parallel_calls': int,
     'prefetch_buffer_size': int,
     'pad_lengths_to_eight': bool,
     'pad_vocab_to_eight': bool,
     'seed_tokens': str,
     'shuffle_buffer_size': int,
     'processed_data_folder': str,
   })
Ejemplo n.º 7
0
 def get_optional_params():
   return dict(DataLayer.get_optional_params(), **{
     'rand_start': bool,
     'small': bool,
     'use_targets': bool,
     'delimiter': str,
     'map_parallel_calls': int,
     'prefetch_buffer_size': int,
     'pad_lengths_to_eight': bool,
     'pad_vocab_to_eight': bool,
     'shuffle_buffer_size': int,
     'data_root': str,
     'binary': bool,
     'num_classes': int,
     'get_stats': bool,
   })
Ejemplo n.º 8
0
 def get_optional_params():
   return dict(
       DataLayer.get_optional_params(), **{
           'pad_to': int,
           'mag_power': int,
           'pad_EOS': bool,
           'pad_value': float,
           'feature_normalize_mean': float,
           'feature_normalize_std': float,
           'trim': bool,
           'data_min': None,
           'duration_min': int,
           'duration_max': int,
           'mel_type': ['slaney', 'htk'],
           "exp_mag": bool
       }
   )
Ejemplo n.º 9
0
 def get_optional_params():
     return dict(DataLayer.get_optional_params(),
                 **{"dataset_location": str})
Ejemplo n.º 10
0
 def get_optional_params():
     return dict(DataLayer.get_optional_params(), **{
         "cache_data": bool,
         "augment_data": bool
     })