ML2TransformerApp / model.py

Commit History

app.py interface, made functions more independent, ensemble, working prototype
96feb73

dkoshman commited on

beam search
29bcc5f

dkoshman commited on

moved constants to separate file, organized tokenizer
c308f77

dkoshman commited on

dedicated generate.py script
41a34cd

dkoshman commited on

noam lr scheduler, shared weight between embedding and generator
e932abd

dkoshman commited on

lr logger
57273ba

dkoshman commited on

added callback on hook, decoder, image logger, tried tuning
4f4785c

dkoshman commited on

tuned cli, added tuner
c2ef1c6

dkoshman commited on

fixed nans, now trains properly, but still hangs when finishing multiprocessing
9cdd1c7

dkoshman commited on

working backend
2a394f6

dkoshman commited on

big changes: transformer, pytorch lightning, argparser
fb8db0f

dkoshman commited on

removed data, added tex preloading
e949d7b

dkoshman commited on

image embedding and encoding
c29b35f

dkoshman commited on

data_preprocessing, base train script
6e82d4a

dkoshman commited on