Skip to content

Commit

Permalink
Fix (examples/a2q): updating file header
Browse files Browse the repository at this point in the history
  • Loading branch information
i-colbert committed Feb 6, 2024
1 parent f3d6fb3 commit 28c74f7
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions src/brevitas_examples/imagenet_classification/a2q/resnet.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
# Copyright (C) 2024, Advanced Micro Devices, Inc. All rights reserved.
# SPDX-License-Identifier: BSD-3-Clause
# Copyright (c) 2024, Advanced Micro Devices, Inc.
# Copyright (c) 2017, liukuang
# All rights reserved.
# SPDX-License-Identifier: MIT

import torch.nn as nn
import torch.nn.functional as F
Expand All @@ -21,6 +23,8 @@ def weight_init(layer):


class BasicBlock(nn.Module):
"""Basic block architecture modified for CIFAR10.
Adapted from https://github.com/kuangliu/pytorch-cifar"""
expansion = 1

def __init__(self, in_planes: int, planes: int, stride: int = 1):
Expand Down Expand Up @@ -55,7 +59,7 @@ def forward(self, x):

class ResNet(nn.Module):
""" ResNet architecture modified for CIFAR10.
Based on https://github.com/kuangliu/pytorch-cifar/blob/master/models/resnet.py"""
Adapted from https://github.com/kuangliu/pytorch-cifar"""

def __init__(self, block_impl, num_blocks, num_classes: int = 10):
super(ResNet, self).__init__()
Expand Down

0 comments on commit 28c74f7

Please sign in to comment.