karthiksagarn commited on
Commit
1a448b4
·
verified ·
1 Parent(s): 5f1a5ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ class Model(nn.Module):
51
  return fmap,self.dp(self.linear1(x_lstm[:,-1,:]))
52
 
53
  class validation_dataset(Dataset):
54
- def __init__(self,video_names,sequence_length=60,transform=None):
55
  self.video_names = video_names
56
  self.transform = transform
57
  self.count = sequence_length
 
51
  return fmap,self.dp(self.linear1(x_lstm[:,-1,:]))
52
 
53
  class validation_dataset(Dataset):
54
+ def __init__(self,video_names,sequence_length=20,transform=None):
55
  self.video_names = video_names
56
  self.transform = transform
57
  self.count = sequence_length