master
Tom Weber 2 years ago
parent 8ca9c8fc0b
commit bc4094ea8c

@ -3,8 +3,8 @@
import re
import os
import torch
from dotd.model import GRU
from dotd.data import get_shakespeare, preprocess
from drama_generator.model import GRU
from drama_generator.data import get_shakespeare, preprocess
def get_header():

@ -5,7 +5,7 @@ import click
from torch.utils.data import DataLoader
from tqdm import tqdm # type: ignore
import numpy as np
from dotd.data import ShakespeareDataset, preprocess, get_shakespeare
from drama_generator.data import ShakespeareDataset, preprocess, get_shakespeare
class GRU(torch.nn.Module):

Loading…
Cancel
Save