Formatting
* Adding black as dev dependency * Formatting pina code * Formatting tests
This commit is contained in:
committed by
Nicola Demo
parent
4c4482b155
commit
42ab1a666b
@@ -1,4 +1,4 @@
|
||||
""" Module for ReducedOrderModelSolver """
|
||||
"""Module for ReducedOrderModelSolver"""
|
||||
|
||||
import torch
|
||||
|
||||
@@ -126,7 +126,7 @@ class ReducedOrderModelSolver(SupervisedSolver):
|
||||
optimizer=optimizer,
|
||||
scheduler=scheduler,
|
||||
weighting=weighting,
|
||||
use_lt=use_lt
|
||||
use_lt=use_lt,
|
||||
)
|
||||
|
||||
# assert reduction object contains encode/ decode
|
||||
@@ -185,4 +185,4 @@ class ReducedOrderModelSolver(SupervisedSolver):
|
||||
reduction_network.decode(encode_repr_reduction_network), output_pts
|
||||
)
|
||||
|
||||
return loss_encode + loss_reconstruction
|
||||
return loss_encode + loss_reconstruction
|
||||
|
||||
Reference in New Issue
Block a user