search

CAMMA-public / rendezvous

A transformer-inspired neural network for surgical action triplet recognition from laparoscopic videos.
Other
24 stars 8 forks source link

TensorFlow 2 code #7

Closed IzzeddinTeeti closed 1 year ago

IzzeddinTeeti commented 2 years ago

Hi, I tried the Tensorflow implementation, but it did not work for me. I copied run.py to tensorflow_2.x and I set the directory to tensorflow_2.x then used python run.py -t -e --data_dir="data folder" --dataset_variant=cholect50-challenge --version=1. Did I do something wrong? Can you please guide me on how to run the TensorFlow implementation.

nwoyecid commented 2 years ago

Please note that you will need to follow the run.py in pytorch to re-implement the training script for tensorflow-2.x. Directly using a training script from another framework will obviously not work as tensorflow handles parameter optimization differently from pytorch.