HHousen / lecture2notes

Undefined name detected PYL-E0602
Bug risk
Critical
3 years ago4 years old
494
495def create_head(out_features):
496    layers = [
497        AdaptiveConcatPool2d(1),  # noqa: F405498        nn.Flatten(),
499        nn.BatchNorm1d(1024),
500        nn.Dropout(0.25),