https://github.com/facebookresearch/pythia
Tip revision: c7cac1b65230b806733e97f1e87393af4a3e60f4 authored by Amanpreet Singh on 29 April 2021, 23:00:03 UTC
[fix] CircleCI GPU tests; sharded DDP tests
[fix] CircleCI GPU tests; sharded DDP tests
Tip revision: c7cac1b
File | Mode | Size |
---|---|---|
__init__.py | -rw-r--r-- | 164 bytes |
attention.py | -rw-r--r-- | 9.9 KB |
bottleneck.py | -rw-r--r-- | 6.4 KB |
decoders.py | -rw-r--r-- | 2.7 KB |
embeddings.py | -rw-r--r-- | 20.5 KB |
encoders.py | -rw-r--r-- | 24.6 KB |
fusions.py | -rw-r--r-- | 24.4 KB |
hf_layers.py | -rw-r--r-- | 18.7 KB |
layers.py | -rw-r--r-- | 26.1 KB |
losses.py | -rw-r--r-- | 25.6 KB |
metrics.py | -rw-r--r-- | 41.6 KB |
optimizers.py | -rw-r--r-- | 3.4 KB |
schedulers.py | -rw-r--r-- | 2.1 KB |