-
I am following your rep to fine tune GPT-J on TPU. When I run "python3 device_train.py --config=YOUR_CONFIG.json --tune-model-path=gs://YOUR-BUCKET/step_383500/" with my bucket name and the config fil…
-
from adabelief_pytorch import AdaBelief安装不了,请问怎么解决
-
trying to import optax and getting an error `AttributeError: module 'jax' has no attribute '_src'` for jax versions > 0.3.17
optax version == 0.1.3
chex version == 0.1.3
```
In [1]: import opt…
-
Hello Lu Lu,
DeepXDE is great and I am trying to use it to deal with a complex **Dam Break** problem, that is:
- Length = 10 m
- Height Left = 0.005 m
- Height Right = 0.001 m
- The dam is …
-
Using the OptaxMinimiser as solver results in " '_Closure' object has no attribute 'init' " whereas the BFGS solver runs without errors.
The objective function uses a custom equinox pytree.
-
### 🐛 Describe the bug
I am compiling a model that involves fourier transform and inverse fourier transform of tensors, which involves complex values after the transform. I am getting the error:
…
-
### Search before asking
- [X] I have searched the YOLOv8 [issues](https://github.com/ultralytics/ultralytics/issues) and found no similar feature requests.
### Description
In the `ultralyt…
-
Hi,
I previously had the optx newton root finding algorithm in operation which used a jnp.where to set a default value when the root_finder couldn't find a solution. It worked to insert the default v…
-
I tested training the same deepfake model on the same hardware using tensorflow-cuda and tensorflow-directml. (my project https://github.com/iperov/DeepFaceLab)
DirectML: avg iter time 626ms
![DML…
-
### 🐛 Describe the bug
when given invalid input, crossentropy fails silently.
```python
import torch.nn.functional as F
device = torch.device("cuda")
ignore_index = 255
b =…
joihn updated
7 months ago