something changed idk what

This commit is contained in:
art 2020-01-08 15:01:33 -06:00
parent 4e50a79614
commit 32bc329e91
2 changed files with 2 additions and 7 deletions

View File

@ -317,11 +317,6 @@ def histo_analysis(hist_data):
@jit(forceobj=True) @jit(forceobj=True)
def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterations = 10000, lr = 0.01, _iterations_ply = 10000, lr_ply = 0.01, power_limit = None): # inputs, outputs expects N-D array def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterations = 10000, lr = 0.01, _iterations_ply = 10000, lr_ply = 0.01, power_limit = None): # inputs, outputs expects N-D array
if power_limit == None:
power_limit = len(outputs)
else:
power_limit += 1
regressions = [] regressions = []
Regression().set_device(device) Regression().set_device(device)
@ -345,8 +340,8 @@ def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterat
plys = [] plys = []
if power_limit == None: if power_limit == None:
power_limit = len(outputs[0]) power_limit = len(outputs[0]) - 1
for i in range(2, power_limit): for i in range(2, power_limit):