import json
import os
import random
from tqdm import tqdm

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import optim
from torch.utils.data import Dataset, DataLoader

root_path = '/data/chs/ICMI_data/ballDrop_flow/'
train_path = f'{root_path}/train'
test_path = f'{root_path}/test'

BATCH_SIZE = 3
EPOCHS = 8

mean_left = np.load('mean_left.npy', allow_pickle=True)
std_left = np.load('std_left.npy', allow_pickle=True)