|
import torch
|
|
|
|
|
|
|
|
EXP_NAME = "IAM-1000"; RESUME = True
|
|
|
|
|
|
|
|
IMG_HEIGHT = 32
|
|
resolution = 16
|
|
batch_size = 8
|
|
NUM_EXAMPLES = 50
|
|
TN_HIDDEN_DIM = 512
|
|
TN_DROPOUT = 0.1
|
|
TN_NHEADS = 8
|
|
TN_DIM_FEEDFORWARD = 512
|
|
TN_ENC_LAYERS = 1
|
|
TN_DEC_LAYERS = 1
|
|
ALPHABET = 'Only thewigsofrcvdampbkuq.A-210xT5\'MDL,RYHJ"ISPWENj&BC93VGFKz();#:!7U64Q8?+*ZX/%'
|
|
VOCAB_SIZE = len(ALPHABET)
|
|
G_LR = 0.0002
|
|
D_LR = 0.0002
|
|
W_LR = 0.0002
|
|
OCR_LR = 0.0002
|
|
EPOCHS = 100000
|
|
NUM_CRITIC_GOCR_TRAIN = 2
|
|
NUM_CRITIC_DOCR_TRAIN = 1
|
|
NUM_CRITIC_GWL_TRAIN = 2
|
|
NUM_CRITIC_DWL_TRAIN = 1
|
|
NUM_FID_FREQ = 100
|
|
DATASET = ['IAM']
|
|
DATASET_PATHS = {'IAM':'../IAM_32.pickle'}
|
|
NUM_WRITERS = 1000
|
|
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
IS_SEQ = True
|
|
NUM_WORDS = 3
|
|
if not IS_SEQ: NUM_WORDS = NUM_EXAMPLES
|
|
IS_CC = False
|
|
IS_KLD = False
|
|
ADD_NOISE = False
|
|
ALL_CHARS = False
|
|
SAVE_MODEL = 5
|
|
SAVE_MODEL_HISTORY = 100
|
|
|
|
|
