IBM / Grapher

Code that implements efficient knowledge graph extraction from the textual descriptions
Apache License 2.0
135 stars 30 forks source link

Segment fault: core dumped #17

Open yhshu opened 8 months ago

yhshu commented 8 months ago
bash train_gen.sh 

train_gen.sh: line 29: 3411753 Segmentation fault      (core dumped) python main.py --version 2 --default_root_dir output --run train --max_epochs 100 --accelerator gpu --num_nodes 1 --num_data_workers 4 --lr 1e-4 --batch_size 11 --num_sanity_val_steps 0 --fast_dev_run 0 --overfit_batches 0 --limit_train_batches 1.0 --limit_val_batches 1.0 --limit_test_batches 1.0 --accumulate_grad_batches 10 --detect_anomaly True --data_path webnlg-dataset/release_v3.0/en --log_every_n_steps 100 --val_check_interval 1000 --checkpoint_step_frequency 1000 --focal_loss_gamma 3 --dropout_rate 0.5 --num_layers 2 --edges_as_classes 0 --checkpoint_model_id -1

Sorry, I have no idea why this could be a problem. linux, pytorch_lightning 1.8.5, pytorch 1.13 cuda 113, python 3.8

glad4enkonm commented 7 months ago

Have the same problem in this google colab

installed the next py packages

!pip install torch==1.13
!pip install pytorch-lightning==1.8.1
!pip install nervaluate
!pip install datsets transformers[sentencepiece]
!pip install sentencepiece