當前位置: 首頁>>代碼示例>>Python>>正文


Python Model.to方法代碼示例

本文整理匯總了Python中torchbearer.Model.to方法的典型用法代碼示例。如果您正苦於以下問題:Python Model.to方法的具體用法?Python Model.to怎麽用?Python Model.to使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在torchbearer.Model的用法示例。


在下文中一共展示了Model.to方法的3個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: test_to_only_dtype

# 需要導入模塊: from torchbearer import Model [as 別名]
# 或者: from torchbearer.Model import to [as 別名]
    def test_to_only_dtype(self):
        dtype = torch.float16

        torchmodel = torch.nn.Sequential(torch.nn.Linear(1,1))
        torchmodel.to = Mock()
        optimizer = torch.optim.Adam(torchmodel.parameters(), 0.1)
        state_tensor = torch.Tensor([1])
        state_tensor.to = Mock()
        optimizer.state = {'test': {'test': state_tensor}}

        torchbearermodel = Model(torchmodel, optimizer, torch.nn.L1Loss(), [])
        torchbearermodel.to(dtype)

        self.assertTrue(torchmodel.to.call_args[0][0] == dtype)
        self.assertTrue(state_tensor.to.call_args[0][0] == dtype)
開發者ID:little1tow,項目名稱:torchbearer,代碼行數:17,代碼來源:test_torchbearer.py

示例2: test_cpu

# 需要導入模塊: from torchbearer import Model [as 別名]
# 或者: from torchbearer.Model import to [as 別名]
    def test_cpu(self):
        torchmodel = torch.nn.Sequential(torch.nn.Linear(1,1))
        torchmodel.load_state_dict = Mock()

        optimizer = torch.optim.SGD(torchmodel.parameters(), 0.1)
        optimizer.load_state_dict = Mock()

        torchbearermodel = Model(torchmodel, optimizer, torch.nn.L1Loss(), [])
        torchbearermodel.to = Mock()
        torchbearermodel.cpu()

        self.assertTrue(torchbearermodel.to.call_args[0][0] == 'cpu')
開發者ID:little1tow,項目名稱:torchbearer,代碼行數:14,代碼來源:test_torchbearer.py

示例3: test_cuda_no_device

# 需要導入模塊: from torchbearer import Model [as 別名]
# 或者: from torchbearer.Model import to [as 別名]
    def test_cuda_no_device(self, device_mock):
        device_mock.return_value = 111

        torchmodel = torch.nn.Sequential(torch.nn.Linear(1,1))
        torchmodel.load_state_dict = Mock()

        optimizer = torch.optim.SGD(torchmodel.parameters(), 0.1)
        optimizer.load_state_dict = Mock()

        torchbearermodel = Model(torchmodel, optimizer, torch.nn.L1Loss(), [])
        torchbearermodel.to = Mock()
        torchbearermodel.cuda()

        self.assertTrue(torchbearermodel.to.call_args[0][0] == 'cuda:' + str(111))
開發者ID:little1tow,項目名稱:torchbearer,代碼行數:16,代碼來源:test_torchbearer.py


注:本文中的torchbearer.Model.to方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。