Giters

CAMMA-public / rendezvous

A transformer-inspired neural network for surgical action triplet recognition from laparoscopic videos.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TensorFlow 2 code

IzzeddinTeeti opened this issue · comments

commented

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.

commented

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.