Tutorials v0.1 (#178)
Tutorial update and small fixes * Tutorials update + Tutorial FNO * Create a metric tracker callback * Update PINN for logging * Update plotter for plotting * Small fix LabelTensor * Small fix FNO --------- Co-authored-by: Dario Coscia <dariocoscia@cli-10-110-13-250.WIFIeduroamSTUD.units.it> Co-authored-by: Dario Coscia <dariocoscia@dhcp-176.eduroam.sissa.it>
This commit is contained in:
committed by
Nicola Demo
parent
939353f517
commit
a9b1bd2826
@@ -63,7 +63,7 @@ class LabelTensor(torch.Tensor):
|
||||
if isinstance(labels, str):
|
||||
labels = [labels]
|
||||
|
||||
if len(labels) != x.shape[1]:
|
||||
if len(labels) != x.shape[-1]:
|
||||
raise ValueError(
|
||||
'the tensor has not the same number of columns of '
|
||||
'the passed labels.'
|
||||
|
||||
Reference in New Issue
Block a user