Describe the bug
当使用较大的模型训练数据时,创建任务会报错

Expected behavior
使用模型如下:
class CNN_FEMNIST(nn.Module):
def __init__(self, only_digits=False):
super(CNN_FEMNIST, self).__init__()
self.conv2d_1 = nn.Conv2d(1, 32, kernel_size=3)
self.max_pooling = nn.MaxPool2d(2, stride=2)
self.conv2d_2 = nn.Conv2d(32, 64, kernel_size=3)
self.dropout_1 = nn.Dropout(0.25)
self.flatten = nn.Flatten()
self.linear_1 = nn.Linear(9216, 128)
self.dropout_2 = nn.Dropout(0.5)
self.linear_2 = nn.Linear(128, 10 if only_digits else 62)
self.relu = nn.ReLU()
# self.softmax = nn.Softmax(dim=1)
def forward(self, x):
x = self.conv2d_1(x)
x = self.relu(x)
x = self.conv2d_2(x)
x = self.relu(x)
x = self.max_pooling(x)
x = self.dropout_1(x)
x = self.flatten(x)
x = self.linear_1(x)
x = self.relu(x)
x = self.dropout_2(x)
x = self.linear_2(x)
# x = self.softmax(x)
return x
Describe the bug

当使用较大的模型训练数据时,创建任务会报错
Expected behavior
使用模型如下: