To run this example, use the following command. python example.py data/pseudo_data.pt data/pseudo_data_test.pt 1