 Matthias Reso
|
c5a382e509
Make tests run on cpu only machines
|
1 éve |
 Hamid Shojanazeri
|
162be4c045
Revert "Flop counter, profiling and GC (#357)"
|
1 éve |
|
Hamid Shojanazeri
|
1a09fb5d27
add logging for setting profiler
|
1 éve |
|
Hamid Shojanazeri
|
71d137c722
Merge branch 'main' into flop_counter_gc
|
1 éve |
|
Hamid Shojanazeri
|
8bf474b455
clean up
|
1 éve |
|
Hamid Shojanazeri
|
19089269d3
add gc
|
1 éve |
|
Hamid Shojanazeri
|
dbfea484c6
Feature : Enable Intel GPU/XPU finetuning and inference (#116)
|
1 éve |
 Beto
|
1f5b202c18
Adding tests for the save_metrics param in the train function
|
1 éve |
|
Beto
|
7474514fe0
Merging with main
|
1 éve |
 gaopengzhi
|
c7d410725b
Merge branch 'main' into grad_clip
|
1 éve |
 Abhilash Majumder
|
4793f0fdf3
Merge branch 'main' into ipex_feature
|
1 éve |
|
gaopengzhi
|
e2797abe9b
Add gradient_clipping and gradient_clipping_threshold parameters
|
1 éve |
|
gaopengzhi
|
bb7c6c1e33
Support FSDP scenario
|
1 éve |
|
gaopengzhi
|
b1d9efd155
Refactor gradient clipping feature
|
2 éve |
|
Beto
|
17d02c3b44
Adding config to conditionally save stats
|
2 éve |
|
Beto
|
b974c87035
Merging latest from main
|
2 éve |
 Jeremy Howard
|
eca8410b32
Use bf16 parameters in bf16 mixed prec
|
2 éve |
|
gaopengzhi
|
04befdef69
Add gradient clipping feature
|
2 éve |
|
Abhilash Majumder
|
11465d6329
remove duplicate import
|
2 éve |
|
Abhilash Majumder
|
6a78b96764
Merge branch 'main' into ipex_feature
|
2 éve |
|
Matthias Reso
|
e8bb7fbabc
Merge remote-tracking branch 'origin/main' into feature/length_based_batch_sampling
|
2 éve |
|
Matthias Reso
|
33925f71e6
Add missing amp context if use_fp16 is enabled
|
2 éve |
|
Hamid Shojanazeri
|
35b394e49f
adding profiler and flop_counter
|
2 éve |
|
Hamid Shojanazeri
|
d56d5c469d
adding flop counter
|
2 éve |
|
Beto
|
92be45b0fe
Adding matplotlib to requirements. Removing import from train_utils
|
2 éve |
|
Beto
|
80a6ac41b5
Adding the rank to the metrics filename to differentiate the run from each GPU
|
2 éve |
|
Beto
|
2f7256918e
Adding support to save fine tuning metrics to json. Adding script to plot saved metrics.
|
2 éve |
|
Matthias Reso
|
10f9367e56
fix missing labels in datasets
|
2 éve |
 hongbo.mo
|
6217635e87
Fix tqdm bar not change length after terminal is resized
|
2 éve |
|
Matthias Reso
|
c33ea3cacb
Fix pbar update
|
2 éve |
