Update train.py

    print(f'Received {ary.size} elements of type {ary.dtype}')

    ## Preprocess training sequences

    #data_path = config['data_path']

    #data_np = np.load(data_path)  # lev1_data

    data_np = ary

    ## Prepare the lev1, lev2 training data, for lev2, we have different strategies based on the mapping_mode