add torch-load script