Skip to content

Commit 09dae2b

Browse files
authored
Update README.md
1 parent 6768d0c commit 09dae2b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# Deep Equilibrium Models
22

3-
**2020/10: A [JAX] version of the DEQ, including JAX implementation of Broyden's method, etc. is available [here](https://github.com/akbir/deq-jax).**
3+
**2020/10: A [JAX](https://github.com/google/jax#quickstart-colab-in-the-cloud) version of the DEQ, including JAX implementation of Broyden's method, etc. is available [here](https://github.com/akbir/deq-jax).**
44

55
**2020/7: A branch `pytorch-1.5` has been created to resolve the DataParallel issue with PyTorch v>=1.5 (see [here](https://github.com/pytorch/pytorch/issues/40457) and [here](https://github.com/huggingface/transformers/pull/4300) for details). Specifically, this is not a DEQ-related issue, but one related to some of the modules DEQ depends on (e.g., adaptive embedding). In PyTorch 1.5, accessing parameters on the replicas is no longer possible. For now, you can use the `pytorch-1.5` to train the model from scratch, but there is no pre-trained model yet following the code change. To run pre-trained models, please still use PyTorch 1.4 and this `master` branch.**
66

0 commit comments

Comments
 (0)