search

29rabhishek / EEG_BiLSTM

1 stars 0 forks source link

optimize logics for GPU processing #5

Open 29rabhishek opened 10 months ago

29rabhishek commented 10 months ago

In train_step we transfer data to GPU but there some part of code which can also be optimized