Commit
•
75208a0
1
Parent(s):
b9ed13a
Upload model
Browse files- config.json +1 -1
- model.py +1 -1
config.json
CHANGED
|
@@ -14,7 +14,7 @@
|
|
| 14 |
"classifier": "default",
|
| 15 |
"crop_pct": 0.875,
|
| 16 |
"device": "cpu",
|
| 17 |
-
"first_conv": "
|
| 18 |
"has_aux": false,
|
| 19 |
"input_channels": 3,
|
| 20 |
"input_size": [
|
|
|
|
| 14 |
"classifier": "default",
|
| 15 |
"crop_pct": 0.875,
|
| 16 |
"device": "cpu",
|
| 17 |
+
"first_conv": "conv2d_1a.conv",
|
| 18 |
"has_aux": false,
|
| 19 |
"input_channels": 3,
|
| 20 |
"input_size": [
|
model.py
CHANGED
|
@@ -15,7 +15,7 @@ class SafeSearchConfig(PretrainedConfig):
|
|
| 15 |
interpolation: str = "bicubic",
|
| 16 |
mean: list = [0.5, 0.5, 0.5],
|
| 17 |
std: list = [0.5, 0.5, 0.5],
|
| 18 |
-
first_conv: str = "
|
| 19 |
classifier: str = "default",
|
| 20 |
has_aux: bool = False,
|
| 21 |
label_offset: int = 0,
|
|
|
|
| 15 |
interpolation: str = "bicubic",
|
| 16 |
mean: list = [0.5, 0.5, 0.5],
|
| 17 |
std: list = [0.5, 0.5, 0.5],
|
| 18 |
+
first_conv: str = "conv2d_1a.conv",
|
| 19 |
classifier: str = "default",
|
| 20 |
has_aux: bool = False,
|
| 21 |
label_offset: int = 0,
|