Visual-Behavior / detr-tensorflow

Tensorflow implementation of DETR : Object Detection with Transformers
MIT License
168 stars 53 forks source link

Issues with Training Coco dataset #47

Open rads96 opened 1 year ago

rads96 commented 1 year ago

I'm getting the following error "[[{{node PyFunc}}]] 0 successful operations. 0 derived errors ignored. [Op:__inference_run_val_step_37449]"

on lines

m_outputs, total_loss, log = run_val_step(model, images, t_bbox, t_class, config) and m_outputs, total_loss, log, gradient_steps = run_train_step(model, images, t_bbox, t_class, optimizers,

in the training.py file when i try to train my coco dataset and pass it through

training.eval(detr, valid_dt, config, coco_class_names, evaluation_step=200) training.fit(detr, train_dt, optimzers, config, epoch_nb=0, class_names=coco_class_names)

Does anyone have an idea about this?

Screenshot 2023-04-18 at 6 34 39 PM