Spaces:
Running
Running
| 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 | |