name 'train_stats' is not defined

yrcong / RelTR

RelTR: Relation Transformer for Scene Graph Generation: https://arxiv.org/abs/2201.11460v2

229 stars 45 forks source link

name 'train_stats' is not defined #41

Open G200005 opened 6 months ago

G200005 commented 6 months ago

When I use single NVIDIA GeForce RTX 4090 training, the error named 'train_stats' is not defined is reported, what is the reason for this, the command I am using is: python main.py --dataset vg --img_folder data/vg/images/ --ann_path data/vg/ --batch_size 2 --output_dir ckpt

kent252 commented 4 months ago

I believe that you should look in the code, the author commented on these lines of "train_stats". Uncomment it will solve the problem

qqxqqbot commented 2 months ago

@G200005 Hi! bro, what is the cuda and torch version do you using in rtx4090, I cann't do it.