forked from ccf-ai-infra/GPUCodeForces
This commit is contained in:
parent
751b24be1b
commit
79fe16bda1
|
|
@ -5,7 +5,6 @@ You have complete freedom to choose the set of operators you want to replace. Yo
|
||||||
Here's an example to show you the syntax of inline embedding custom CUDA operators in torch: The example given architecture is:
|
Here's an example to show you the syntax of inline embedding custom CUDA operators in torch: The example given architecture is:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
# mseloss_torch.py
|
|
||||||
import torch
|
import torch
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
import torch.nn.functional as F
|
import torch.nn.functional as F
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue