Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
sarus-llm / sarus_llm / deepspeed_config.json
Size: Mime:
{
    "steps_per_print": 20,
    "prescale_gradients": false,
    "bf16": {
        "enabled": false
    },
    "fp16": {
        "enabled": false,
        "fp16_master_weights_and_grads": false,
        "loss_scale": 1.0,
        "loss_scale_window": 1000,
        "hysteresis": 2,
        "min_loss_scale": 1,
        "initial_scale_power": 0
    },
    "wall_clock_breakdown": false,
    "zero_optimization": {
        "stage": 1,
        "allgather_partitions": true,
        "reduce_scatter": true,
        "allgather_bucket_size": 50000000,
        "reduce_bucket_size": 50000000,
        "overlap_comm": false,
        "contiguous_gradients": true,
        "cpu_offload": false,
        "stage3_max_live_parameters" : 1e8,
        "stage3_max_reuse_distance" : 1e8,
        "stage3_prefetch_bucket_size" : 1e7
    }
  }