Added total_save_limit feature similar to run_langauge_modeling.py
Added args.total_save_limit in order to save only the last specific checkpoints similar to the feature in run_langauge_modeling.py. This might be helpful for a student like me who has limited space storage quota on the school's remote server.
I feel its not necessary