mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-30 17:52:35 +06:00
96f57c9ccb
* improve memory benchmarking * correct typo * fix current memory * check torch memory allocated * better pytorch function * add total cached gpu memory * add total gpu required * improve torch gpu usage * update memory usage * finalize memory tracing * save intermediate benchmark class * fix conflict * improve benchmark * improve benchmark * finalize * make style * improve benchmarking * correct typo * make train function more flexible * fix csv save * better repr of bytes * better print * fix __repr__ bug * finish plot script * rename plot file * delete csv and small improvements * fix in plot * fix in plot * correct usage of timeit * remove redundant line * remove redundant line * fix bug * add hf parser tests * add versioning and platform info * make style * add gpu information * ensure backward compatibility * finish adding all tests * Update src/transformers/benchmark/benchmark_args.py Co-authored-by: Lysandre Debut <lysandre@huggingface.co> * Update src/transformers/benchmark/benchmark_args_utils.py Co-authored-by: Lysandre Debut <lysandre@huggingface.co> * delete csv files * fix isort ordering * add out of memory handling * add better train memory handling Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
10 lines
144 B
Plaintext
10 lines
144 B
Plaintext
tensorboard
|
|
scikit-learn
|
|
seqeval
|
|
psutil
|
|
sacrebleu
|
|
rouge-score
|
|
tensorflow_datasets
|
|
pytorch-lightning==0.7.3 # April 10, 2020 release
|
|
matplotlib
|