refactored model tests

This commit is contained in:
William Falcon committed 2019-07-24 13:41:28 -04:00
1 parent 24ceafa05c
commit b90841dc3d
2 files changed
+4 -6

No files matched your search

+2 -6
View File
@@ -105,14 +105,10 @@ def main():
checkpoint = ModelCheckpoint(save_dir)
trainer = Trainer(
checkpoint_callback=checkpoint,
progress_bar=True,
experiment=exp,
progress_bar=False,
max_nb_epochs=1,
train_percent_check=0.7,
val_percent_check=0.1,
gpus=[0, 1],
distributed_backend='ddp',
distributed_backend='dp',
use_amp=True
)