From 3be8be228e911798eb791166d7eac0ec7f4e2d5a Mon Sep 17 00:00:00 2001
From: zzaawy <63528364+zzaawy@users.noreply.github.com>
Date: Mon, 20 Jan 2025 23:27:53 -0500
Subject: [PATCH] Update adv_image.py

---
 adv_image.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/adv_image.py b/adv_image.py
index 2e90304..143cd09 100644
--- a/adv_image.py
+++ b/adv_image.py
@@ -22,6 +22,8 @@ def weights_init(m):
     - BatchNorm layers: Weights from N(1.0, 0.02), biases=0
     """
     classname = m.__class__.__name__
+    # Check if the layer is a convolutional layer
+    # The find() method returns -1 if 'Conv' is not found in the class name
     if classname.find('Conv') != -1:
         nn.init.normal_(m.weight.data, 0.0, 0.02)
     elif classname.find('BatchNorm') != -1: