Spaces:
Build error
Build error
Update app.py
Browse files
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=
|
| 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
|