text: import torch
  from sequence import EventSeq, ControlSeq
  
  #pylint: disable=E1101
  
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
  
  model = {
      'init_dim': 32,
      'event_dim': EventSeq.dim(),
      'control_dim': ControlSeq.dim(),
      'hidden_dim': 512,
      'gru_layers': 3,
      'gru_dropout': 0.3,
  }
  
  train = {
      'learning_rate': 0.001,    
      'batch_size': 64,
      'window_size': 200,
      'stride_size': 10,
      'use_transposition': False,
      'control_ratio': 1.0,
      'teacher_forcing_ratio': 1.0
  }
  