frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[functorch] excise remaining functorch imports from examples (#91282)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/91282
Approved by: https://github.com/zou3519
This commit is contained in:
samdow
2023-01-05 11:29:14 -05:00
committed by PyTorch MergeBot
parent 071756c9cf
commit 39524f20de
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
functorch/examples/lennard_jones/lennard_jones.py
View File

@ -5,7 +5,7 @@
import torch import torch
from torch import nn from torch import nn
from torch.nn.functional import mse_loss from torch.nn.functional import mse_loss
from functorch import jacrev, vmap from torch.func import jacrev, vmap
sigma = 0.5 sigma = 0.5
epsilon = 4. epsilon = 4.

2
functorch/examples/maml_regression/evjang_transforms.py
View File

@ -2,7 +2,7 @@
# (https://github.com/ericjang/maml-jax). # (https://github.com/ericjang/maml-jax).
# We translated his implementation from JAX to PyTorch. # We translated his implementation from JAX to PyTorch.
from functorch import grad, vmap from torch.func import grad, vmap
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import math import math
import torch import torch