Hello, I am trying to implement this architecture (Figure 6 | COVID-19 Pandemic Forecasting Using CNN-LSTM: A Hybrid Approach) using torch.nn.LSTM but I am not sure what 200 units LSTM corresponds to in nn.LSTM parameters. I would guess it is num_layers but then how will hidden_size be determined?