Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file added examples/figs/fig_01.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added examples/figs/manifold.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
6 changes: 1 addition & 5 deletions src/neuromancer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,20 +6,16 @@

import torch
from torch.optim.lr_scheduler import ReduceLROnPlateau

import numpy as np
import wandb
import lightning.pytorch as pl
import lightning.pytorch as pl

from neuromancer.loggers import BasicLogger
from neuromancer.problem import Problem
from neuromancer.callbacks import Callback
from neuromancer.problem import LitProblem
from neuromancer.dataset import LitDataModule

from lightning.pytorch.callbacks import ModelCheckpoint
from lightning.pytorch.callbacks.early_stopping import EarlyStopping
from lightning.pytorch.loggers import WandbLogger



Expand Down
1 change: 0 additions & 1 deletion src/neuromancer/utils.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@

import torch
import functools
import lightning.pytorch as pl
from collections import OrderedDict

def handle_device_placement(func):
Expand Down