trax
trax copied to clipboard
google
Trax — Deep Learning with Clear Code and Speed
### Description Unable to train a reformer model using tensorflow-numpy due to incompatable types ### Environment information ``` OS: Colaboratory $ pip freeze | grep trax trax==1.3.6 (same issue in...
### Description `ReversibleSelect` seems to mess up JAX's backtracing/JIT compilation. In the code provided below, we define a simple (non reversible) model which (1) splits the inputs; (2) does something...
https://github.com/google/trax/blob/b8acc0e8f5c52a5ac25fd98a7a196f13b5fc1f69/trax/layers/base.py#L127 See here https://docs.python.org/3/library/inspect.html#inspect.currentframe
### Description I'm trying to build a simple sentiment analysis model using the code from the tutorial you provided here: https://trax-ml.readthedocs.io/en/latest/notebooks/trax_intro.html However I'm not able to use certain metrics, although...
### Description ... ### Environment information ``` OS: $ pip freeze | grep trax # your output here $ pip freeze | grep tensor # your output here $ pip...
### Description Wen I run this machine_translation.ipynb,it not work # Error logs: TypeError: reshape total size must be unchanged, got new_sizes (1, 1, 1, 128) for shape (1, 1024, 128)....
### Description I'm building subwords dictionary for Polish language and receive error when running text_encoder_build_subword.py. To run text_encoder_build_subword.py there should at the last line of code ```app.run(main)``` instead of ```app.run()```...
### Description When attempting to train a Reformer with LSH Attention with n_hashes > 1 on a TPU, training will get stuck, and the trainer is not able to complete...
### when I run the official code,but the error "NameError: name 'base' is not defined" happens. The code follow is the same as official code. import numpy as np import...
### Description Imagenet64 dataset from tensor2tensor used in this gin config: https://github.com/google/trax/blob/master/trax/supervised/configs/reformer_imagenet64.gin seems to have some loading issues. I tried to run this config on Google Colab: https://colab.research.google.com/drive/1ysEQYOaIspHPBVu6S9jOxc7BkE2oDrh0 and ran...
