import json
import numpy as np
import os
import random
from tqdm import tqdm

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import optim
from torch.utils.data import Dataset, DataLoader, random_split
from torchvision import transforms, utils

total_body_points = 9
classes = ['Pass', 'No_Pass', 'Raise_Lower']
frame_size = (9, 2)

EPOCHS = 10
BATCH_SIZE = 3

data_path = '../alphapose_json_March_2020/'
data_path

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
device