Unexpected data transfer via PCI when training model on GPU
hghcomphys opened this issue · comments
The structure data and model weights exist on GPU (as device array) therefore there shouldn't be a heavy PCI-throughput during the training.
Part of this issue was related to structure.elements property which has to map atom_type to element.