TypeError: train_eval_model() got an unexpected keyword argument 'use_tpu_wrapper'

google-research / tensor2robot

Distributed machine learning infrastructure for large-scale robotics research

https://ai.google/research/teams/brain/robotics/

Apache License 2.0

537 stars 116 forks source link

TypeError: train_eval_model() got an unexpected keyword argument 'use_tpu_wrapper' #15

Open aRnaZ91 opened 2 years ago

aRnaZ91 commented 2 years ago

Hi, I have tried to run QT-opt from my PC last week but I have not been able to make it work yet. I am using conda to be able to get the right version of every packages needed to run the project (and especially to get tensorflow 1 and avoid the migration to tensorflow 2). Though, even if I have the packages of requirement.txt with the right versions, I am still getting errors like showed on the screenshot (problem with use_tpu_wrapper). Do you have any ideas of how to fix these errors to run the code in the right way? Thank you in advance for your consideration. QTrobot2 QT2robot1

anandhupvr commented 1 year ago

Hi, by removing or commenting use_tpu_wrapper=params['use_tpu_wrapper'] argument from t2r_test_fixture.py solve this error.

inside random_train_model function


          t2r_model=tf_model,
          input_generator_train=input_generator,
          max_train_steps=params['max_train_steps'],
          model_dir=params['model_dir'],
          use_tpu_wrapper=params['use_tpu_wrapper'])```