* Adds pretrained IDs directly in the tests * Fix tests * Fix tests * Review!
num_hidden_layers=2
Styling