Non-trainable params:
0
(0.00 B)