-
Notifications
You must be signed in to change notification settings - Fork 0
/
settings.py
35 lines (27 loc) · 1.18 KB
/
settings.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import json
import torch
from torch.autograd import Variable
from utils_dir.utils import Singleton
class CnfSettings(metaclass=Singleton):
def __init__(self, hyperparameters={'cuda': False}):
self.hyperparameters = hyperparameters
self.LongTensor = torch.cuda.LongTensor if hyperparameters['cuda'] else torch.LongTensor
self.FloatTensor = torch.cuda.FloatTensor if hyperparameters['cuda'] else torch.FloatTensor
self.ByteTensor = torch.cuda.ByteTensor if hyperparameters['cuda'] else torch.ByteTensor
self.expand_dim_const = Variable(self.zeros(1), requires_grad=False)
def __getitem__(self, key):
return self.hyperparameters[key]
def zeros(self,size):
rc = torch.zeros(size)
return rc.cuda() if self.hyperparameters['cuda'] else rc
def cudaize(self, func, *args, **kwargs):
rc = func(*args,**kwargs)
return rc.cuda() if self.hyperparameters['cuda'] else rc
def cudaize_var(self, var):
return var.cuda() if self.hyperparameters['cuda'] else var
def update_from_dict(self, update_dict):
for k in update_dict.keys():
self.hyperparameters[k] = update_dict[k]
def update_from_json(self, fname):
with open(fname, 'r') as fp:
self.update_from_dict(json.load(fp))