From 07c3c1b47f74afb0db3787a7cc8b33e23c852225 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 21 Oct 2024 13:50:31 +0200
Subject: [PATCH] datasets
---
model_custom.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/model_custom.py b/model_custom.py
index f829002..82c1516 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -3,6 +3,7 @@ import torch.nn as nn
import torch
from tape.models.modeling_bert import ProteinBertModel
+
class PermuteLayer(nn.Module):
def __init__(self, dims):
super().__init__()
@@ -102,6 +103,7 @@ class Model_Common_Transformer(nn.Module):
def forward(self, seq, charge):
print(seq.size(),charge.size())
print(seq, charge)
+ print(torch.cuda.mem_get_info())
meta_ohe = torch.nn.functional.one_hot(charge - 1, self.charge_max).float()
seq_emb = torch.nn.functional.one_hot(seq, self.nb_aa).float()
print(seq_emb.shape)
--
GitLab