Training of assembly sequence generation

[Friedrich-Graaf]

We are trying to train the network on our data to generate assembly sequences for parts, but the results dont look promising.
The paper to the repository mentions more loss criteria like the kendall loss, which is (as far as i see it) not part of this code base.
Do you have suggestions what to try out to improve the performance of the assembly sequence predictions?