From 73c379378c9310579d3c911ea927998f3888d290 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 15 Oct 2024 15:11:27 +0200
Subject: [PATCH] seq_length args
---
common_dataset.py | 2 +-
model_custom.py | 1 -
2 files changed, 1 insertion(+), 2 deletions(-)
diff --git a/common_dataset.py b/common_dataset.py
index dbed634..b2fca94 100644
--- a/common_dataset.py
+++ b/common_dataset.py
@@ -152,8 +152,8 @@ class Common_Dataset(Dataset):
def load_data(path_train, path_val, path_test, batch_size, length, pad=False, convert=False, vocab = 'unmod'):
print('Loading data')
- data_train = pd.read_pickle(path_train)
data_val = pd.read_pickle(path_val)
+ data_train = pd.read_pickle(path_train)
data_test = pd.read_pickle(path_test)
train = Common_Dataset(data_train, length, pad, convert, vocab)
test = Common_Dataset(data_val, length, pad, convert, vocab)
diff --git a/model_custom.py b/model_custom.py
index 96cfc64..eddc158 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -1,7 +1,6 @@
import math
import torch.nn as nn
import torch
-from tape import TAPETokenizer
from tape.models.modeling_bert import ProteinBertModel
class PermuteLayer(nn.Module):
--
GitLab