haonanhe/Transformer-in-spike-trains

main_trains.py is the main file.
transformer.py contains models.