MNCJihun commited on
Commit
512a750
·
1 Parent(s): 2f77bb4

add grad false part

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ model.fc = nn.Sequential(
43
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
44
  model.load_state_dict(torch.load(MODEL_PATH))
45
  model.eval()
 
 
46
 
47
  inputs = gr.inputs.Image()
48
  outputs = gr.outputs.Label(num_top_classes=1)
 
43
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
44
  model.load_state_dict(torch.load(MODEL_PATH))
45
  model.eval()
46
+ for _, p in model.named_parameters():
47
+ p.requires_grad = False
48
 
49
  inputs = gr.inputs.Image()
50
  outputs = gr.outputs.Label(num_top_classes=1)