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
from torch import nn
from torch.nn.functional import mse_loss
from functorch import jacrev, vmap
from torch.func import jacrev, vmap
sigma = 0.5
epsilon = 4.

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

@ -2,7 +2,7 @@
# (https://github.com/ericjang/maml-jax).
# 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 math
import torch