fix loss normalization for ddp training

haoneng.lhn

2023-12-08 acb9a0fec8d8a4dabeedcbb8e08c26f66d7083f0

1
2
3
4
5
6
7
from modelscope.hub.snapshot_download import snapshot_download
import sys
 
if __name__ == "__main__":
    model_tag = sys.argv[1]
    local_model_dir = sys.argv[2]
    model_dir = snapshot_download(model_tag, cache_dir=local_model_dir, revision='1.0.0')