def _get_rnn_model(self, input_size:int):
rnn_args = {
'input_size':input_size,
'hidden_size':256,
'num_layers':2
}
return RNN(rnn_args, 'lstm'), rnn_args['hidden_size']
def _get_rnn_model(self, input_size: int):
""" Builder method to get RNN instace."""
rnn_args = {
'input_size': input_size,
'hidden_size': 512,
'num_layers': 2
}
return RNN(rnn_args, 'lstm'), rnn_args['hidden_size']
def _get_rnn_model(self, input_size: int):
""" Builder method to instantiate an RNN object."""
rnn_args = {
'input_size': input_size,
'hidden_size': 256,
'num_layers': 2,
'batch_first': True
}
return RNN(rnn_args, 'gru'), rnn_args['hidden_size']
def _get_rnn_model(self, input_size:int):
""" Builder method to get RNN instace."""
rnn_args = {
'input_size': input_size,
'hidden_size': 64,
'num_layers': 2,
'batch_first':True
}
return RNN(rnn_args, 'gru'), rnn_args['hidden_size']