Fixed git lfs issue
This commit is contained in:
1
.gitattributes
vendored
1
.gitattributes
vendored
@@ -1 +0,0 @@
|
|||||||
*.csv filter=lfs diff=lfs merge=lfs -text
|
|
||||||
@@ -59,11 +59,11 @@ else:
|
|||||||
quantiles = eval(quantiles)
|
quantiles = eval(quantiles)
|
||||||
|
|
||||||
model_parameters = {
|
model_parameters = {
|
||||||
"learning_rate": 0.001,
|
"learning_rate": 0.0001,
|
||||||
"hidden_size": 512,
|
"hidden_size": 256,
|
||||||
"num_layers": 4,
|
"num_layers": 4,
|
||||||
"dropout": 0.2,
|
"dropout": 0.2,
|
||||||
"time_feature_embedding": 8,
|
"time_feature_embedding": 16,
|
||||||
}
|
}
|
||||||
|
|
||||||
model_parameters = task.connect(model_parameters, name="model_parameters")
|
model_parameters = task.connect(model_parameters, name="model_parameters")
|
||||||
@@ -96,7 +96,7 @@ trainer = AutoRegressiveQuantileTrainer(
|
|||||||
trainer.add_metrics_to_track(
|
trainer.add_metrics_to_track(
|
||||||
[PinballLoss(quantiles), MSELoss(), L1Loss(), CRPSLoss(quantiles)]
|
[PinballLoss(quantiles), MSELoss(), L1Loss(), CRPSLoss(quantiles)]
|
||||||
)
|
)
|
||||||
trainer.early_stopping(patience=15)
|
trainer.early_stopping(patience=30)
|
||||||
trainer.plot_every(5)
|
trainer.plot_every(5)
|
||||||
trainer.train(task=task, epochs=epochs, remotely=True)
|
trainer.train(task=task, epochs=epochs, remotely=True)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user