csarofeen

from torch._C._nvfuser import Fusion, FusionCache, FusionDefinition, DataType
import torch
import torch.nn.functional as F
import functorch
from functorch.compile import memory_efficient_fusion
from copy import deepcopy
from typing import List
import time
import functools
import random

csarofeen

import time

import torch
from torch.autograd import Variable
import torchvision.models as models

import torch.backends.cudnn as cudnn
cudnn.benchmark = True

net = models.vgg19().cuda()

csarofeen

import torch
import torch.nn as nn
from torch.autograd import Variable

torch.backends.cudnn.enabled=True

seq_size = 2
hidden_size = 3
inp_size = 4
batch_size = 5

csarofeen

import torch
import torch.nn as nn
from torch.autograd import Variable
import itertools

seq_size = 16
batch_size = 32
inp_size = 64
hidden_size = 128

csarofeen

from __future__ import print_function
import argparse
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torch.autograd import Variable
import time