mirror of https://github.com/hpcaitech/ColossalAI
6 lines
92 B
Python
6 lines
92 B
Python
|
import torch
|
||
|
|
import torch.nn as nn
|
||
|
|
|
||
|
|
# list of inplace operations
|
||
|
|
INPLACE_MODULE = [nn.ReLU]
|