usage: main.py [--nn] [--layers] [--lr] [--batch_size] [--attention]
               [--bidirectional] [--supervised] [--results]

optional arguments:
  --nn            neural network to use - linear / lstm / gru
  --layers        number and size of layers - linear:   neurons_layer_1 neurons_layer_2 ...
                                            - lstm/gru: hidden_dimension stacked_layers
  --lr            learning rate - float
  --batch_size    batch size - int
  --attention     option to include attention layers for lstm/gru
  --bidirectional option to include bidirectionality for lstm/gru
  --supervised    option to train on ground truth parameters
  --results       option to perform evaluation on network using different SNR (must be trained first)