DeepSpeed
Solve tensor vs numpy dtype conflicts in data efficiency map-reduce.
#5108
Merged
Merged
Solve tensor vs numpy dtype conflicts in data efficiency map-reduce. #5108
conglongli
merged
15 commits into
deepspeedai:master
from
assert_map_and_reduce_types
added assert of torch vs numpy types
14f2bbe1
removed extra line at the end of the file
222bdc1b
bm-synth
changed the title
Merge branch 'master' into assert_map_and_reduce_types
8f7d8c65
missing metric_dtypes in custom_map_update
bf0bae06
Merge branch 'assert_map_and_reduce_types' of github.com:bm-synth/Dee…
5d4e0f1b
accept numpy and tensor return values in metric_function. added type …
67ac38db
bm-synth
changed the title
Merge branch 'master' into assert_map_and_reduce_types
f6b059bd
bm-synth
changed the title
Update indexed_dataset.py
7192bc2d
Merge branch 'master' into assert_map_and_reduce_types
675394ad
Merge branch 'master' into assert_map_and_reduce_types
1999ce7c
pre-commit formatting
118d631d
pre-commit hooks
42641564
format
ba65f2aa
Merge branch 'master' into assert_map_and_reduce_types
a2d02398
Merge branch 'master' into assert_map_and_reduce_types
8f44d974
conglongli
approved these changes
on 2024-02-15
Login to write a write a comment.
Login via GitHub