mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-10-14 23:58:11 +08:00
fix
Former-commit-id: 53a74cbc3afec58b36c2265e080061bcdf702f98
This commit is contained in:
parent
3021b31cf3
commit
527ba2eb2e
@ -138,7 +138,7 @@ def compute_mfu(
|
|||||||
result = json.load(f)
|
result = json.load(f)
|
||||||
|
|
||||||
mfu_value = (
|
mfu_value = (
|
||||||
result["train_samples_per_second"]
|
result["train_steps_per_second"]
|
||||||
* compute_model_flops(model_name_or_path, batch_size, seq_length)
|
* compute_model_flops(model_name_or_path, batch_size, seq_length)
|
||||||
/ compute_device_flops()
|
/ compute_device_flops()
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user