In [ ]:

    

import argparse
import os
import shutil
import nltk
from collections import defaultdict

import torch
import torch.nn as nn
from torch.autograd import Variable
from torch.utils.data.dataloader import DataLoader

from dpp_nets.utils.language import Vocabulary, BeerDataset, simple_collate, custom_collate
from dpp_nets.layers.layers import ChunkTrainer, ChunkTrainerRel
from dpp_nets.layers.layers import KernelVar, PredNet, ReinforceSampler