Packages

flyte

1.16.1
2.0.20 2.0.19 2.0.17 2.0.16 2.0.15 2.0.14 2.0.13 2.0.12 2.0.11 2.0.9 2.0.8 2.0.7 2.0.6 2.0.5 2.0.2 2.0.1 2.0.0 2.0.0-alpha16 2.0.0-alpha15 2.0.0-alpha14 2.0.0-alpha13 2.0.0-alpha12 2.0.0-alpha11 2.0.0-alpha10 1.16.1 1.5.22 1.5.21-rc1

Elixir files generated from Flyte's protobuf files using protobuf-elixir.

HexDocs

Current section

Files

Jump to
Readme
27 Versions
2 Dependencies
Dependants
Files
Activity
flyte lib flyteidl plugins kubeflow pytorch.pb.ex
Raw

lib/flyteidl/plugins/kubeflow/pytorch.pb.ex 

defmodule Flyteidl.Plugins.Kubeflow.ElasticConfig do

  @moduledoc false



  use Protobuf, syntax: :proto3, protoc_gen_elixir_version: "0.12.0"



  field :rdzv_backend, 1, type: :string, json_name: "rdzvBackend"

  field :min_replicas, 2, type: :int32, json_name: "minReplicas"

  field :max_replicas, 3, type: :int32, json_name: "maxReplicas"

  field :nproc_per_node, 4, type: :int32, json_name: "nprocPerNode"

  field :max_restarts, 5, type: :int32, json_name: "maxRestarts"

end



defmodule Flyteidl.Plugins.Kubeflow.DistributedPyTorchTrainingTask do

  @moduledoc false



  use Protobuf, syntax: :proto3, protoc_gen_elixir_version: "0.12.0"



  field :worker_replicas, 1,

    type: Flyteidl.Plugins.Kubeflow.DistributedPyTorchTrainingReplicaSpec,

    json_name: "workerReplicas"



  field :master_replicas, 2,

    type: Flyteidl.Plugins.Kubeflow.DistributedPyTorchTrainingReplicaSpec,

    json_name: "masterReplicas"



  field :run_policy, 3, type: Flyteidl.Plugins.Kubeflow.RunPolicy, json_name: "runPolicy"



  field :elastic_config, 4,

    type: Flyteidl.Plugins.Kubeflow.ElasticConfig,

    json_name: "elasticConfig"

end



defmodule Flyteidl.Plugins.Kubeflow.DistributedPyTorchTrainingReplicaSpec do

  @moduledoc false



  use Protobuf, syntax: :proto3, protoc_gen_elixir_version: "0.12.0"



  field :replicas, 1, type: :int32

  field :image, 2, type: :string

  field :resources, 3, type: Flyteidl.Core.Resources



  field :restart_policy, 4,

    type: Flyteidl.Plugins.Kubeflow.RestartPolicy,

    json_name: "restartPolicy",

    enum: true

end