DSSM(
(embed_user): Embedding(6, 2, padding_idx=5)
(embed_item): Embedding(6, 2, padding_idx=5)
(embed_feat): ModuleDict(
(a): Embedding(3, 2, padding_idx=2)
(b): Embedding(3, 2, padding_idx=2)
)
(fcu1): Linear(in_features=4, out_features=5, bias=True)
(fcu2): Linear(in_features=5, out_features=3, bias=True)
(fcu3): Linear(in_features=3, out_features=2, bias=True)
(fci1): Linear(in_features=4, out_features=5, bias=True)
(fci2): Linear(in_features=5, out_features=3, bias=True)
(fci3): Linear(in_features=3, out_features=2, bias=True)
(bnu1): BatchNorm1d(5, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(bnu2): BatchNorm1d(3, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(bni1): BatchNorm1d(5, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
(bni2): BatchNorm1d(3, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
)