Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
4f38d06
1
Parent(s):
ba4f1af
Update evaclip/modified_resnet.py
Browse files
evaclip/modified_resnet.py
CHANGED
|
@@ -4,7 +4,7 @@ import torch
|
|
| 4 |
from torch import nn
|
| 5 |
from torch.nn import functional as F
|
| 6 |
|
| 7 |
-
from
|
| 8 |
|
| 9 |
|
| 10 |
class Bottleneck(nn.Module):
|
|
|
|
| 4 |
from torch import nn
|
| 5 |
from torch.nn import functional as F
|
| 6 |
|
| 7 |
+
from evaclip.utils import freeze_batch_norm_2d
|
| 8 |
|
| 9 |
|
| 10 |
class Bottleneck(nn.Module):
|