forked from ccf-ai-infra/GPUCodeForces
fixes switchable #12
This commit is contained in:
parent
9a38eca08a
commit
af453c19b0
|
|
@ -10,7 +10,6 @@ import torch
|
|||
import torch.nn as nn
|
||||
import torch.nn.functional as F
|
||||
|
||||
# 定义输入尺寸 (N, C, H, W)
|
||||
N, C, H, W = 16, 64, 32, 32
|
||||
EPS = 1e-5
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue