我上了这个课:
NUM_IMGS, HEIGHT, WIDTH = images.shape
NUM_CLASSES = np.unique(labels).size
D_H = 100
NUM_OPT_STEPS = 100
class TwoLayerNet(torch.nn.Module):
def __init__(self, HEIGHT*WIDTH, D_H, NUM_CLASSES):
其中高度,宽度,D_H和num_classes在上面几行中定义。 但是,当我运行脚本时,出现以下错误:
File "<ipython-input-43-cf9af12418a7>", line 2
def __init__(self, HEIGHT*WIDTH, D_H, NUM_CLASSES):
^
SyntaxError: invalid syntax
这可能是一个愚蠢的问题,但是此语法如何无效?我需要在init中定义产品还是将产品作为单个变量传递?