Skip to content
Snippets Groups Projects
Commit a49eb721 authored by Schneider Leo's avatar Schneider Leo
Browse files

dataset rain ISA

parent a9276df0
No related branches found
No related tags found
No related merge requests found
...@@ -11,7 +11,7 @@ from config_common import load_args ...@@ -11,7 +11,7 @@ from config_common import load_args
from common_dataset import load_data from common_dataset import load_data
from dataloader import load_data from dataloader import load_data
from loss import masked_cos_sim, distance, masked_spectral_angle from loss import masked_cos_sim, distance, masked_spectral_angle
from model_custom import Model_Common_Transformer, Model_Common_Transformer_TAPE from model_custom import Model_Common_Transformer
from model import RT_pred_model_self_attention_multi from model import RT_pred_model_self_attention_multi
...@@ -25,7 +25,6 @@ def train(model, data_train, epoch, optimizer, criterion_rt, criterion_intensity ...@@ -25,7 +25,6 @@ def train(model, data_train, epoch, optimizer, criterion_rt, criterion_intensity
for param in model.parameters(): for param in model.parameters():
param.requires_grad = True param.requires_grad = True
if forward == 'both': if forward == 'both':
print(data_train.dataset.data['Sequence'])
for seq, charge, rt, intensity in data_train: for seq, charge, rt, intensity in data_train:
rt, intensity = rt.float(), intensity.float() rt, intensity = rt.float(), intensity.float()
if torch.cuda.is_available(): if torch.cuda.is_available():
...@@ -201,7 +200,7 @@ def main(args): ...@@ -201,7 +200,7 @@ def main(args):
data_train, data_val, data_test = common_dataset.load_data(path_train=args.dataset_train, data_train, data_val, data_test = common_dataset.load_data(path_train=args.dataset_train,
path_val=args.dataset_val, path_val=args.dataset_val,
path_test=args.dataset_test, path_test=args.dataset_test,
batch_size=args.batch_size, length=25, pad = False, convert=True, vocab='iapuc') batch_size=args.batch_size, length=25, pad = True, convert=True, vocab='iapuc')
elif args.forward == 'rt': elif args.forward == 'rt':
data_train, data_val, data_test = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test], data_train, data_val, data_test = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test],
batch_size=args.batch_size, length=25) batch_size=args.batch_size, length=25)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment