Python map_location Examples

Programming Language: Python

Namespace/Package Name: d3rlpy.algos.torch.utility

Method/Function: map_location

Examples at hotexamples.com: 3

Python map_location - 3 examples found. These are the top rated real world Python examples of d3rlpy.algos.torch.utility.map_location extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

def test_map_location_with_cuda():
    fn = map_location('cuda:0')
    dummy = Mock()
    dummy.cuda = Mock()

    fn(dummy, '')

    dummy.cuda.assert_called_with('cuda:0')

Example #2

Show file

def test_map_location_with_cpu():
    assert map_location('cpu:0') == 'cpu'

Example #3

Show file

 def load_model(self, fname):
     chkpt = torch.load(fname, map_location=map_location(self.device))
     set_state_dict(self, chkpt)