-
- Downloads
Flop counter, profiling and GC (#357)
Showing
- src/llama_recipes/configs/training.py 3 additions, 0 deletionssrc/llama_recipes/configs/training.py
- src/llama_recipes/finetuning.py 5 additions, 6 deletionssrc/llama_recipes/finetuning.py
- src/llama_recipes/utils/__init__.py 2 additions, 1 deletionsrc/llama_recipes/utils/__init__.py
- src/llama_recipes/utils/tflop_counter.py 464 additions, 0 deletionssrc/llama_recipes/utils/tflop_counter.py
- src/llama_recipes/utils/train_utils.py 96 additions, 2 deletionssrc/llama_recipes/utils/train_utils.py
Loading
Please register or sign in to comment