Class: Torch::Distributed::TCPStore

Inherits:

Object

• Object
• Torch::Distributed::TCPStore

Defined in:

lib/torch/distributed.rb

Class Method Summary

• .new(host, port, world_size, is_master, wait_for_workers: true, timeout: DEFAULT_TIMEOUT) ⇒ Object

Class Method Details

.new(host, port, world_size, is_master, wait_for_workers: true, timeout: DEFAULT_TIMEOUT) ⇒ Object

# File 'lib/torch/distributed.rb', line 443

def self.new(host, port, world_size, is_master, wait_for_workers: true, timeout: DEFAULT_TIMEOUT)
  Torch::Distributed._create_tcp_store(host, port, world_size, is_master, (timeout * 1000).to_i, wait_for_workers)
end