Skip to content

Integrate torch script generation with training script

choucl requested to merge feat/integrate-torch-script into wip/redesign-multi-mlp

I have updated the gen_script.py file to include a gen_script function, which is used for generating a torch script immediately after the model training.

Merge request reports