Python 2 compatibility. (#5887)

2026-07-11 10:49:42 +08:00 · 2019-10-10 19:09:25 -07:00
parent c3b2ae26c5
commit 523c764c25
12 changed files with 12 additions and 12 deletions
@@ -113,7 +113,7 @@ def dataset_creators(use_cuda):
 import torch.optim as optim


-class Network():
+class Network(object):
    def __init__(self, lr=0.01, momentum=0.5):
        use_cuda = torch.cuda.is_available()
        self.device = device = torch.device("cuda" if use_cuda else "cpu")