Error 2

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
File D:\ProgramData\Miniconda_3.9\envs\rnn-sample-py3.9\lib\site-packages\torchinfo\torchinfo.py:295, in forward_pass(model, x, batch_dim, cache_forward_pass, device, mode, **kwargs)
    294 if isinstance(x, (list, tuple)):
--> 295     _ = model(*x, **kwargs)
    296 elif isinstance(x, dict):

File D:\ProgramData\Miniconda_3.9\envs\rnn-sample-py3.9\lib\site-packages\torch\nn\modules\module.py:1518, in Module._wrapped_call_impl(self, *args, **kwargs)
   1517 else:
-> 1518     return self._call_impl(*args, **kwargs)

File D:\ProgramData\Miniconda_3.9\envs\rnn-sample-py3.9\lib\site-packages\torch\nn\modules\module.py:1568, in Module._call_impl(self, *args, **kwargs)
   1566     args = bw_hook.setup_input_hook(args)
-> 1568 result = forward_call(*args, **kwargs)
   1569 if _global_forward_hooks or self._forward_hooks:

TypeError: forward() missing 1 required positional argument: 'buffer_in'

The above exception was the direct cause of the following exception:

RuntimeError                              Traceback (most recent call last)
Cell In[21], line 2
      1 from torchinfo import summary
----> 2 summary(gps_rnn, input_size=(batch_size, 32, 14, 2))

File D:\ProgramData\Miniconda_3.9\envs\rnn-sample-py3.9\lib\site-packages\torchinfo\torchinfo.py:223, in summary(model, input_size, input_data, batch_dim, cache_forward_pass, col_names, col_width, depth, device, dtypes, mode, row_settings, verbose, **kwargs)
    216 validate_user_params(
    217     input_data, input_size, columns, col_width, device, dtypes, verbose
    218 )
    220 x, correct_input_size = process_input(
    221     input_data, input_size, batch_dim, device, dtypes
    222 )
--> 223 summary_list = forward_pass(
    224     model, x, batch_dim, cache_forward_pass, device, model_mode, **kwargs
    225 )
    226 formatting = FormattingOptions(depth, verbose, columns, col_width, rows)
    227 results = ModelStatistics(
    228     summary_list, correct_input_size, get_total_memory_used(x), formatting
    229 )

File D:\ProgramData\Miniconda_3.9\envs\rnn-sample-py3.9\lib\site-packages\torchinfo\torchinfo.py:304, in forward_pass(model, x, batch_dim, cache_forward_pass, device, mode, **kwargs)
    302 except Exception as e:
    303     executed_layers = [layer for layer in summary_list if layer.executed]
--> 304     raise RuntimeError(
    305         "Failed to run torchinfo. See above stack traces for more details. "
    306         f"Executed layers up to: {executed_layers}"
    307     ) from e
    308 finally:
    309     if hooks:

RuntimeError: Failed to run torchinfo. See above stack traces for more details. Executed layers up to: []