mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-12-26 17:49:09 +00:00
analysis.py v 1.1.11.009
This commit is contained in:
parent
2e8ad61224
commit
1fa0fd6d7a
Binary file not shown.
@ -7,10 +7,12 @@
|
||||
# current benchmark of optimization: 1.33 times faster
|
||||
# setup:
|
||||
|
||||
__version__ = "1.1.11.008"
|
||||
__version__ = "1.1.11.009"
|
||||
|
||||
# changelog should be viewed using print(analysis.__changelog__)
|
||||
__changelog__ = """changelog:
|
||||
1.1.11.009:
|
||||
- bug fixes
|
||||
1.1.11.008:
|
||||
- bug fixes
|
||||
1.1.11.007:
|
||||
@ -342,6 +344,10 @@ def regression(device, inputs, outputs, args, loss = torch.nn.MSELoss(), _iterat
|
||||
|
||||
plys = []
|
||||
|
||||
if power_limit == None:
|
||||
|
||||
power_limit = len(outputs[0])
|
||||
|
||||
for i in range(2, power_limit):
|
||||
|
||||
model = Regression().SGDTrain(Regression.PolyRegKernel(len(inputs),i), torch.tensor(inputs).to(torch.float).to(device), torch.tensor(outputs).to(torch.float).to(device), iterations=_iterations_ply * 10 ** i, learning_rate=lr_ply * 10 ** -i, return_losses=True)
|
||||
|
Loading…
Reference in New Issue
Block a user