device=cuda:0, depth=4, d_hidden=256, epochs=100, seed=42