Jisnoo Dev Pal
Results
3
comments of
Jisnoo Dev Pal
Import torch.nn Wrap the model in nn.DataParallel() like this below, classifier = torch.nn.DataParallel(model.get_model(num_class, normal_channel=args.use_normals)) In terminal, export CUDA_VISIBLE_DEVICES= Now run train_classification.py
@singjc @timosachsenberg I would be happy to help you on this!
Added the note. You can review the [PR](https://github.com/OpenMS/pyopenms_viz/pull/70). Let me know if I need to make any other changes.