From c2df044d5c2f68591b21367646926eab14e2a9fd Mon Sep 17 00:00:00 2001 From: "vejvarm@freya" Date: Mon, 27 Nov 2023 15:46:38 +0900 Subject: [PATCH] refactor: clean up imports --- build_vocabs.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/build_vocabs.py b/build_vocabs.py index b9b9ed9..d8f94b3 100644 --- a/build_vocabs.py +++ b/build_vocabs.py @@ -1,22 +1,14 @@ -import time import random import logging -from functools import partial import numpy as np -import torch.optim -from tqdm import tqdm - -from model import CARTON -from dataset import CSQADataset, collate_fn -from torch.utils.data import DataLoader, SequentialSampler, BatchSampler, RandomSampler -from torch.utils.tensorboard import SummaryWriter -from utils import (NoamOpt, AverageMeter, SingleTaskLoss, MultiTaskLoss, save_checkpoint, init_weights) +from dataset import CSQADataset from helpers import setup_logger from constants import * from args import get_parser + parser = get_parser() args = parser.parse_args()