chess/transformer.py from mfinzi/OMGchess

chess/transformer.py

1 hr

    def __init__(self,hidden_dim,vocab_size,num_blocks,inner_dim,num_heads=4,embed_dim=16,dropout=0):

Found in chess/transformer.py - About 50 mins to fix

    def __init__(self,hidden_dim,inner_dim,num_heads=4,dropout=0,rel_pos=True):

Found in chess/transformer.py - About 35 mins to fix

mfinzi/OMGchess