search

jolibrain / wheatley

Next-generation scheduling problem solver based on GNNs and Reinforcement Learning
Other
44 stars 11 forks source link

feat(logs): use torchinfo to plot model trace #45

Closed pierrot-lc closed 1 year ago

pierrot-lc commented 1 year ago

The plot is activated by default. Use the arg --skip_model_trace to deactivate it.