louishenrifranc/attention

Error when running

Opened this issue · 1 comments

Experiment will be run without saving current version of the code/config
Running Task Data Generation
100%|######################################################################################| 200000/200000 [00:22<00:00, 8875.19it/s]
100%|########################################################################################| 30000/30000 [00:03<00:00, 8611.42it/s]
Run Task Data Generation
Run Attention Traininig
INFO:tensorflow:Using config: {'_model_dir': 'transformer_output/model', '_tf_random_seed': None, '_save_summary_steps': 500, '_save_checkpoints_steps': 500, '_save_checkpoints_secs': None, '_session_config': None, '_keep_checkpoint_max': 5, '_keep_checkpoint_every_n_hours': 10000, '_log_step_count_steps': 100, '_train_distribute': None, '_device_fn': None, '_service': None, '_cluster_spec': <tensorflow.python.training.server_lib.ClusterSpec object at 0x7facaa92ce48>, '_task_type': 'worker', '_task_id': 0, '_global_id_in_cluster': 0, '_master': '', '_evaluation_master': '', '_is_chief': True, '_num_ps_replicas': 0, '_num_worker_replicas': 1, 'environment': None}
Traceback (most recent call last):
  File "attention/services/attention_train/attentiontrain.py", line 83, in <module>
    TrainAttention(**args).main()
  File "attention/services/attention_train/attentiontrain.py", line 56, in main
    self.train_and_evaluate(model=model)
  File "attention/services/attention_train/attentiontrain.py", line 71, in train_and_evaluate
    validation_answer_filename=os.path.join(self.datasets.valid_data_dir, "answer.txt"))
  File "/attention/attention/algorithms/transformer/transformer.py", line 68, in train_and_evaluate
    max_sequence_len=train_params["max_sequence_len"])
KeyError: 'max_sequence_len'

same problem, anyone knows how to solve it?