Integrate torch script generation with training script
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.
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.