Update tester.py

    parser.add_argument('--gpu_id', type=int, default=1)

    # Dataset options

    parser.add_argument('--dataset_path', type=str, default='')

    parser.add_argument('--data_nums', type=int, default=0)

    parser.add_argument('--batch_size', type=int, default=1)

    parser.add_argument('--num_workers', type=int, default=4)

    parser.add_argument('--T', type=int, default=20)

    if not os.path.exists(args.checkpoints_path):

        raise ValueError('Unknown checkpoints path: {}'.format(checkpoints_path))

    if args.data_nums == 0:

        raise ValueError('Wrong data numbers: {}'.format(args.data_nums))

    if args.checkpoints_file == '':

        args.checkpoints_file = 'sdim{}_ddim{}_cdim{}_hdim{}_winsize{}_T{}_l{}'.format(

                                 args.s_dims,

                         args.l,

                         args.start_epoch)

    kpi_value_test = KpiReader(args.dataset_path, args.data_nums)

    kpi_value_test = KpiReader(args.dataset_path)

    test_loader = torch.utils.data.DataLoader(kpi_value_test, 

                                              batch_size  = args.batch_size,