30 lines
993 B
Rust
30 lines
993 B
Rust
//! Quick smoke test for new architectures
|
|
|
|
#[cfg(test)]
|
|
mod quick_tests {
|
|
use rtx_vision::architectures::*;
|
|
use rtx_vision::{Device, Tensor};
|
|
|
|
#[test]
|
|
fn test_basic_functionality() {
|
|
let device = Device::cpu();
|
|
|
|
// Test ResNet
|
|
let resnet = ResNet::new(ResNetConfig::resnet18()).unwrap();
|
|
let input = Tensor::randn(&[1, 3, 224, 224], &device).unwrap();
|
|
assert!(resnet.forward(&input).is_ok());
|
|
|
|
// Test VGG
|
|
let vgg = VGG::new(VGGConfig::vgg11()).unwrap();
|
|
assert!(vgg.forward(&input).is_ok());
|
|
|
|
// Test DenseNet (commented out - needs stub implementation)
|
|
// let densenet = DenseNet::new(DenseNetConfig::densenet121()).unwrap();
|
|
// assert!(densenet.forward(&input).is_ok());
|
|
|
|
// Test MobileNet (commented out - needs stub implementation)
|
|
// let mobilenet = MobileNet::new(MobileNetConfig::default()).unwrap();
|
|
// assert!(mobilenet.forward(&input).is_ok());
|
|
}
|
|
}
|