Prepare run 3
This commit is contained in:
File diff suppressed because one or more lines are too long
@@ -3,7 +3,7 @@ import pathlib
|
||||
import pandas as pd
|
||||
from aiRNN import dataloader, models, losses
|
||||
|
||||
MODE = 2
|
||||
MODE = 3
|
||||
DEVICE = "cuda:2"
|
||||
|
||||
|
||||
@@ -233,7 +233,7 @@ def run_mode_3():
|
||||
cls = models.ThreeInputLSTM
|
||||
hidden_size = 16
|
||||
rnn_size = 64
|
||||
hidden_layers = 1
|
||||
hidden_layers = 0
|
||||
rnn_layers = 3
|
||||
rnn_dropout = 0.0
|
||||
|
||||
|
||||
Reference in New Issue
Block a user