folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 Format Python code with psf/black

Browse Source
This commit is contained in:
ndem0
2024-04-04 12:42:10 +00:00
committed by Nicola Demo
parent 5c50906771
commit 50fb3b731c
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pina/callbacks/adaptive_refinment_callbacks.py
View File

@@ -56,11 +56,12 @@ class R3Refinement(Callback):
elif precision == "32-true": elif precision == "32-true":
precision = torch.float32 precision = torch.float32
else: else:
raise RuntimeError("Currently R3Refinement is only implemented " raise RuntimeError(
"Currently R3Refinement is only implemented "
"for precision '32-true' and '64-true', set " "for precision '32-true' and '64-true', set "
"Trainer precision to match one of the " "Trainer precision to match one of the "
"available precisions.") "available precisions."
)
# compute residual # compute residual
res_loss = {} res_loss = {}