Good morning guys.
I am trying to run a bunch of (complicated) code but I keep getting the following weird error:
INFO:root:Random state initialized with seed 28794055  
INFO:root:Ani1 will be loaded...
INFO:root:create splits...
INFO:root:load data...
INFO:root:calculate statistics...
INFO:root:cached statistics was loaded...
INFO:root:training...
Traceback (most recent call last):
  File "schnetpack_ani1.py", line 296, in <module>
    train(args, model, train_loader, val_loader, device)
  File "schnetpack_ani1.py", line 156, in train
    trainer.train(device)
  File "/home/kim.a.nicoli/Projects/Schnetpack_release/src/schnetpack/train.py", line 175, in train
    raise e
  File "/home/kim.a.nicoli/Projects/Schnetpack_release/src/schnetpack/train.py", line 118, in train
    loss = self.loss_fn(train_batch, result)
  File "schnetpack_ani1.py", line 149, in loss
    diff = batch[args.property] - result[0]
==TypeError: sub() received an invalid combination of arguments - got (map), but expected one of:
 * (Tensor other, float alpha)
 * (float other, float alpha)==

What I don’t get is that if I insert some print statement at the level where it crashes I get that the types of the objects are e.g. Tensors, and not map.
Does any of you have any suggestion on what to look for or what might be the problem? I didn’t find too much about issues with map type.
Thanks in advance for the help