https://github.com/facebookresearch/pythia
Tip revision: 54c8b110aadaad1124083276531b1f12f42f5889 authored by Ronghang Hu on 01 June 2021, 21:41:22 UTC
[fix] fix TPU checkpoint saving, parameter broadcasting, and misc stuff
[fix] fix TPU checkpoint saving, parameter broadcasting, and misc stuff
Tip revision: 54c8b11
| 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-- | 27.9 KB |
| metrics.py | -rw-r--r-- | 44.9 KB |
| optimizers.py | -rw-r--r-- | 3.4 KB |
| schedulers.py | -rw-r--r-- | 2.1 KB |
